Lista de frameworks compatíveis

Esta página lista os frameworks compatíveis e os pacotes constituintes deles para os seguintes serviços:

Contêineres pré-criados para treinamento personalizado
Contêineres pré-criados para previsões e explicações.
Contêineres pré-criados para o Ray na Vertex AI
Pipelines da Vertex AI
Notebooks do Vertex AI Workbench gerenciados pelo usuário

Para uma explicação detalhada da política de suporte do framework, consulte Política de suporte do framework da Vertex AI.

Frameworks compatíveis com o treinamento personalizado

A Vertex AI usa imagens de contêiner para configurar recursos de nuvem para atender às suas solicitações de treinamento. As seguintes versões são compatíveis com contêineres pré-criados de treinamento personalizado:

TensorFlow

Versão do framework de MLs Aceleradores compatíveis (e a versão CUDA, se aplicável) Fim do patch e da data de suporte Fim da disponibilidade Imagens compatíveis

2.15 (Python 3.10)

Somente TPU

23 de outubro de 2024

23 de outubro de 2025

us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.15.x (patch mais recente) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Other Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.14 (Python 3.10)

Somente CPU

26 de setembro de 2024

26 de setembro de 2025

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.14.x (patch mais recent) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.14 (Python 3.10)

GPU (CUDA 11.x)

26 de setembro de 2024

26 de setembro de 2025

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.14.x (patch mais recent) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.13 (Python 3.10)

Somente CPU

5 de julho de 2024

5 de julho de 2025

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.13.x (patch mais recente) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.13 (Python 3.10)

GPU (CUDA 11.x)

5 de julho de 2024

5 de julho de 2025

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.13.x (patch mais recente) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.12 (Python 3.10)

Somente CPU

30 de junho de 2024

30 de junho de 2025

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.12.x (patch mais recente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.12 (Python 3.10)

GPU (CUDA 11.x)

30 de junho de 2024

30 de junho de 2025

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.12.x (patch mais recente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.12 (Python 3.8)

Somente TPU

30 de junho de 2024

30 de junho de 2025

us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest
europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest
asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.12.x (patch mais recente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.11 (Python 3.10)

Somente CPU

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.11.x (patch mais recente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.11 (Python 3.10)

GPU (CUDA 11.x)

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.11.x (patch mais recente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.11 (Python 3.7)

Somente CPU

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.11.x (patch mais recente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.11 (Python 3.7)

GPU (CUDA 11.x)

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.11.x (patch mais recente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.9 (Python 3.10)

Somente CPU

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.9.x (patch mais recente) sympy 1.10.1 xgboost 1.6.1 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.5.0 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.9 (Python 3.10)

GPU (CUDA 11.x)

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.9.x (patch mais recente) sympy 1.10.1 xgboost 1.6.1 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.5.0 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.9 (Python 3.7)

Somente CPU

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.9.x (patch mais recente) sympy 1.10.1 xgboost 1.6.1 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.5.0 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.9 (Python 3.7)

GPU (CUDA 11.x)

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.9.x (patch mais recente) sympy 1.10.1 xgboost 1.6.1 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.5.0 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.8 (Python 3.10)

Somente CPU

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.8.x (patch mais recente) sympy 1.9 xgboost 1.5.2 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.3 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.8 (Python 3.10)

GPU (CUDA 11.x)

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.8.x (patch mais recente) sympy 1.9 xgboost 1.5.2 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.3 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.8 (Python 3.7)

Somente CPU

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.8.x (patch mais recente) sympy 1.9 xgboost 1.5.2 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.3 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.8 (Python 3.7)

GPU (CUDA 11.x)

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
tensorflow 2.8.x (patch mais recente) sympy 1.9 xgboost 1.5.2 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.3 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Outros Dependências do Deep Learning Containers

2.7 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

tensorflow 2.7.x (patch mais recente)
sympy 1.9
xgboost 1.5.0
python-json-logger 2.0.2
WebOb 1.8.7
Paste 3.5.0
webapp2 3.0.0b1
mock 4.0.3
google-cloud-resource-manager 1.3.1
cloudml-hypertune
Other Dependências do Deep Learning Containers

curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv

2.7 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

2.6 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

tensorflow 2.6.x (patch mais recente)
sympy 1.8
xgboost 1.4.2
python-json-logger 2.0.2
WebOb 1.8.7
Paste 3.5.0
webapp2 3.0.0b1
mock 4.0.3
google-cloud-resource-manager 1.0.2
cloudml-hypertune
Outros Dependências do Deep Learning Containers

2.6 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

2.5 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

tensorflow 2.5.x (patch mais recente)
sympy 1.8
xgboost 1.4.0
PyYAML 3.13
python-json-logger 2.0.1
WebOb 1.8.7
Paste 3.5.0
webapp2 3.0.0b1
mock 4.0.3
google-cloud-resource-manager 0.30.3
cloudml-hypertune
Outras dependências do Deep Learning Containers

2.5 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

2.4 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

tensorflow 2.4.x (patch mais recente)
numpy 1.19.4
pandas 1.1.5
scipy 1.5.4
scikit-learn 0.24.0
sympy 1.7.1
statsmodels 0.12.1
xgboost 1.3.1
oauth2client 4.1.3
httplib2 0.15.0
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 3.13
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.0
python-json-logger 2.0.1
wheel 0.36.2
WebOb 1.8.6
Paste 3.5.0
tornado 5.1.1
grpcio 1.32.0
requests 2.25.1
webapp2 3.0.0b1
mock 4.0.3
google-cloud-bigquery 2.6.1
google-cloud-bigtable 1.6.1
google-cloud-datastore 2.1.0
google-cloud-logging 1.15.0
google-cloud-pubsub 2.2.0
google-cloud-resource-manager 0.30.3
google-cloud-storage 1.35.0
joblib 1.0.0
cloudml-hypertune
psutil 5.8.0

2.4 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

2.3 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

tensorflow 2.3.x (patch mais recente)
numpy 1.18.5
pandas 1.1.3
scipy 1.4.1
scikit-learn 0.23.2
sympy 1.6.2
statsmodels 0.12.0
xgboost 1.2.1
oauth2client 4.1.3
httplib2 0.15.0
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 3.13
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.0
python-json-logger 2.0.1
wheel 0.35.1
WebOb 1.8.6
Paste 3.5.0
tornado 5.1.1
grpcio 1.33.1
requests 2.24.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 2.2.0
google-cloud-bigtable 1.5.1
google-cloud-datastore 1.15.3
google-cloud-logging 1.15.0
google-cloud-pubsub 2.1.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.32.0
joblib 0.17.0
cloudml-hypertune
psutil 5.7.2

2.3 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

2.2 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

tensorflow 2.2.x (patch mais recente)
numpy 1.18.5
pandas 1.0.4
scipy 1.4.1
scikit-learn 0.23.1
sympy 1.6
statsmodels 0.11.1
xgboost 1.1.1
oauth2client 4.1.3
httplib2 0.18.1
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 5.3.1
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.3
python-json-logger 0.1.11
wheel 0.34.2
WebOb 1.8.6
Paste 3.4.1
tornado 5.1.1
grpcio 1.29.0
requests 2.23.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 1.25.0
google-cloud-bigtable 1.2.1
google-cloud-datastore 1.12.0
google-cloud-logging 1.15.0
google-cloud-pubsub 1.6.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.29.0
joblib 0.15.1
cloudml-hypertune
psutil 5.7.0

2.2 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

2.1 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

tensorflow 2.1.0
numpy 1.18.0
pandas 0.25.3
scipy 1.4.1
scikit-learn 0.22.1
sympy 1.5
statsmodels 0.10.2
xgboost 0.90
oauth2client 4.1.3
httplib2 0.15.0
python-dateutil 2.8.1
six 1.13.0
future 0.18.2
PyYAML 5.2
wrapt 1.11.2
crcmod 1.7
google-api-python-client 1.7.11
python-json-logger 0.1.11
wheel 0.33.6
WebOb 1.8.5
Paste 3.2.3
tornado 5.1.1
grpcio 1.26.0
requests 2.22.0
webapp2 3.0.0b1
mock 3.0.5
google-cloud-bigquery 1.23.1
google-cloud-bigtable 1.2.0
google-cloud-datastore 1.10.0
google-cloud-logging 1.14.0
google-cloud-pubsub 1.1.0
google-cloud-resource-manager 0.30.0
google-cloud-storage 1.23.0
joblib 0.14.1
cloudml-hypertune
psutil 5.7.0

2.1 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

1.15 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest
europe-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest
asia-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

tensorflow 1.15.0
numpy 1.16.5
pandas 0.24.2
scipy 1.2.2
scikit-learn 0.20.4
sympy 1.4
statsmodels 0.10.1
xgboost 0.82
oauth2client 4.1.3
httplib2 0.12.0
python-dateutil 2.7.5
argparse 1.4.0
six 1.12.0
future 0.17.1
PyYAML 3.13
wrapt 1.11.1
crcmod 1.7
google-api-python-client 1.7.8
python-json-logger 0.1.10
wheel 0.32.3
WebOb 1.8.5
Paste 3.0.6
tornado 5.1.1
grpcio 1.18.0
requests 2.19.0
webapp2 3.0.0b1
mock 2.0.0
google-cloud-bigquery 1.20.0
google-cloud-bigtable 1.0.0
google-cloud-datastore 1.9.0
google-cloud-logging 1.12.1
google-cloud-pubsub 1.0.0
google-cloud-resource-manager 0.29.2
google-cloud-storage 1.19.1
joblib 0.13.0
cloudml-hypertune
psutil 5.7.0

1.15 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest
europe-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest
asia-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

PyTorch

Versão do framework de MLs Aceleradores compatíveis (e a versão CUDA, se aplicável) Fim do patch e da data de suporte Fim da disponibilidade Imagens compatíveis

2.3 (Python 3.10)

Somente CPU

24 de abril de 2025

24 de abril de 2026

us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-3.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-3.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-3.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.3.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

2.3 (Python 3.10)

Somente GPU (CUDA 12.x)

24 de abril de 2025

24 de abril de 2026

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-3.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-3.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-3.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.3.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

2.2 (Python 3.10)

Somente CPU

30 de janeiro de 2024

23 de outubro de 2026

us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-2.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-2.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-2.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.2.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

2.2 (Python 3.10)

Somente GPU (CUDA 12.x)

30 de janeiro de 2024

30 de janeiro de 2025

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.2.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

2.1 (Python 3.10)

Somente CPU

4 de outubro de 2024

4 de outubro de 2025

us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-1.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-1.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-1.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.1.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

2.1 (Python 3.10)

Somente GPU (CUDA 12.x)

4 de outubro de 2024

4 de outubro de 2025

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-1.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-1.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-1.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.1.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

2.1 (Python 3.8)

Somente TPU

23 de outubro de 2024

23 de outubro de 2025

us-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-1.cp310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-1.cp310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-1.cp310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.1.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

2.0 (Python 3.10)

Somente CPU

15 de março de 2024

15 de março de 2025

us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-0.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-0.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-0.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.0.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

2.0 (Python 3.10)

Somente GPU (CUDA 11.x)

15 de março de 2024

15 de março de 2025

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-0.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-0.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-0.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.0.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

2.0 (Python 3.8)

Somente TPU

15 de maio de 2024

15 de maio de 2025

us-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-0:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-0:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-0:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 2.0.x (patch mais recente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Outro Dependências do Deep Learning Containers	curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

1.13 (Python 3.10)

CPU ou GPU (CUDA 11.x)

15 de maio de 2024

15 de maio de 2025

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 1.13.x (patch mais recente) absl-py 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2

1.13 (Python 3.7)

CPU or GPU (CUDA 11.x)

8 de dezembro de 2023

8 de dezembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 1.13.x (patch mais recente) absl-py 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 cloudml-hypertune Outros Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2

1.12 (Python 3.10)

CPU or GPU (CUDA 11.x)

15 de maio de 2024

15 de maio de 2025

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 1.12.x (patch mais recente) absl-py 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 cloudml-hypertune Outro Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2

1.12 (Python 3.7)

CPU or GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 1.12.x (patch mais recente) absl-py 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 cloudml-hypertune Outro Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2

1.11 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 1.11.x (patch mais recente) absl-py 1.0.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Outro Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2

1.11 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 1.11.x (patch mais recente) absl-py 1.0.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Outro Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2

1.10 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 1.10.x (patch mais recente) absl-py 1.0.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Other Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2

1.10 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
PyTorch 1.10.x (patch mais recente) absl-py 1.0.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Other Dependências do Deep Learning Containers	ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2

1.9 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

PyTorch 1.9.x (patch mais recente)
absl-py 0.13.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Outros Dependências do Deep Learning Containers

1.9 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

PyTorch 1.9.x (patch mais recente)
absl-py 0.13.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Outros Dependências do Deep Learning Containers

1.7 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest

Dependências incluídas

Dependências do Deep Learning Containers

1.7 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest

Dependências incluídas

Dependências do Deep Learning Containers

1.6 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest

Dependências incluídas

Dependências do Deep Learning Containers

1.6 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest

Dependências incluídas

Dependências do Deep Learning Containers

1.4 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest

Dependências incluídas

Dependências do Deep Learning Containers

1.4 (Python 3.7)

GPU (CUDA 11.x)

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest
europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest
asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest

Dependências incluídas

Dependências do Deep Learning Containers

scikit-learn

Versão do framework de MLs Aceleradores compatíveis (e a versão CUDA, se aplicável) Fim do patch e da data de suporte Fim da disponibilidade Imagens compatíveis

1.0 (Python 3.10)

Somente CPU

30 de junho de 2024

30 de junho de 2025

us-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest
europe-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest
asia-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

numpy 1.18.5
pandas 1.0.4
scipy 1.4.1
scikit-learn 1.0
sympy 1.6
statsmodels 0.11.1
xgboost 1.6.2
oauth2client 4.1.3
httplib2 0.18.1
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 5.3.1
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.3
python-json-logger 0.1.11
wheel 0.34.2
WebOb 1.8.6
Paste 3.4.1
tornado 5.1.1
grpcio 1.29.0
requests 2.23.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 1.25.0
google-cloud-bigtable 1.2.1
google-cloud-datastore 1.12.0
google-cloud-logging 1.15.0
google-cloud-pubsub 1.6.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.29.0
joblib 0.15.1
cloudml-hypertune

curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.10
python3-dev
python3.10-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv

1.0 (Python 3.10)

GPU (CUDA 11.x)

30 de junho de 2024

30 de junho de 2025

us-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-0:latest
europe-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-0:latest
asia-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-0:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.10
python3-dev
python3.10-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv

0.23 (Python 3.7)

Somente CPU

1 de setembro de 2023

1 de setembro de 2024

us-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest
europe-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest
asia-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

numpy 1.18.5
pandas 1.0.4
scipy 1.4.1
scikit-learn 0.23.1
sympy 1.6
statsmodels 0.11.1
xgboost 1.1.1
oauth2client 4.1.3
httplib2 0.18.1
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 5.3.1
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.3
python-json-logger 0.1.11
wheel 0.34.2
WebOb 1.8.6
Paste 3.4.1
tornado 5.1.1
grpcio 1.29.0
requests 2.23.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 1.25.0
google-cloud-bigtable 1.2.1
google-cloud-datastore 1.12.0
google-cloud-logging 1.15.0
google-cloud-pubsub 1.6.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.29.0
joblib 0.15.1
cloudml-hypertune

XGBoost

Versão do framework de MLs Aceleradores compatíveis (e a versão CUDA, se aplicável) Fim do patch e da data de suporte Fim da disponibilidade Imagens compatíveis

1.6 (Python 3.10)

Somente CPU

30 de junho de 2024

30 de junho de 2025

us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-6:latest
europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-6:latest
asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-6:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

numpy 1.18.5
pandas 1.0.4
scipy 1.4.1
scikit-learn 0.23.1
sympy 1.6
statsmodels 0.11.1
xgboost 1.6.2
oauth2client 4.1.3
httplib2 0.18.1
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 5.3.1
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.3
python-json-logger 0.1.11
wheel 0.34.2
WebOb 1.8.6
Paste 3.4.1
tornado 5.1.1
grpcio 1.29.0
requests 2.23.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 1.25.0
google-cloud-bigtable 1.2.1
google-cloud-datastore 1.12.0
google-cloud-logging 1.15.0
google-cloud-pubsub 1.6.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.29.0
joblib 0.15.1
cloudml-hypertune

curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.10
python3-dev
python3.10-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv

1.6 (Python 3.10)

GPU (CUDA 11.x)

30 de junho de 2024

30 de junho de 2025

us-docker.pkg.dev/vertex-ai/training/xgboost-gpu.1-6:latest
europe-docker.pkg.dev/vertex-ai/training/xgboost-gpu.1-6:latest
asia-docker.pkg.dev/vertex-ai/training/xgboost-gpu.1-6:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.10
python3-dev
python3.10-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv

1.1 (Python 3.7)

Somente CPU

15 de novembro de 2023

15 de novembro de 2024

us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest
europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest
asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest

Dependências incluídas

Pacotes PyPI

Pacotes Ubuntu

Frameworks compatíveis com previsões e explicações

A Vertex AI usa imagens de contêiner para configurar recursos de nuvem para atender às solicitações de previsão. As versões a seguir são compatíveis com contêineres pré-criados para previsões e explicações:

TensorFlow

Versão do framework de MLs	Aceleradores compatíveis (e a versão CUDA, se aplicável)	Fim do patch e da data de suporte	Fim da disponibilidade	Imagens compatíveis
2,13	Somente CPU	28 de novembro de 2024	28 de novembro de 2025	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-13:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-13:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-13:latest`
2,13	GPU (CUDA 12.x)	28 de novembro de 2024	28 de novembro de 2025	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-13:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-13:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-13:latest`
2,12	Somente CPU	30 de junho de 2024	30 de junho de 2025	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-12:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-12:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-12:latest`
2,12	GPU (CUDA 11.x)	30 de junho de 2024	30 de junho de 2025	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-12:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-12:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-12:latest`
2.11	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-11:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-11:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-11:latest`
2.11	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-11:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-11:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-11:latest`
2.10	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-10:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-10:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-10:latest`
2.10	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-10:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-10:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-10:latest`
2.9	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-9:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-9:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-9:latest`
2.9	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-9:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-9:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-9:latest`
2.8	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-8:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-8:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-8:latest`
2.8	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-8:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-8:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-8:latest`
2.7	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-7:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-7:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-7:latest`
2.7	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-7:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-7:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-7:latest`
2,6	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-6:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-6:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-6:latest`
2,6	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-6:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-6:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-6:latest`
2.5	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-5:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-5:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-5:latest`
2.5	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-5:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-5:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-5:latest`
2,4	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-4:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-4:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-4:latest`
2,4	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-4:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-4:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-4:latest`
2.3	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-3:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-3:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-3:latest`
2.3	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-3:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-3:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-3:latest`
2.2	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-2:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-2:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-2:latest`
2.2	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest`
2.1	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest`
2.1	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest`
1.15	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest`
1.15	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/tf-gpu.1-15:latest` `europe-docker.pkg.dev/vertex-ai/prediction/tf-gpu.1-15:latest` `asia-docker.pkg.dev/vertex-ai/prediction/tf-gpu.1-15:latest`

Ambiente de execução otimizado do TensorFlow

As imagens de contêiner a seguir usam o ambiente de execução otimizado do TensorFlow. Para mais informações, consulte Usar o ambiente de execução otimizado do TensorFlow.

Versão do framework de MLs	Aceleradores compatíveis (e a versão CUDA, se aplicável)	Fim do patch e da data de suporte	Fim da disponibilidade	Imagens compatíveis
noturno	Somente CPU	Não aplicável	Não aplicável	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.nightly:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.nightly:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.nightly:latest`
noturno	GPU (CUDA 12.x)	Não aplicável	Não aplicável	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.nightly:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.nightly:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.nightly:latest`
noturno	Cloud TPU	Não aplicável	Não aplicável	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.nightly:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.nightly:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.nightly:latest`
2,16	Somente CPU	26 de abril de 2024	26 de abril de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-16:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-16:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-16:latest`
2,16	GPU (CUDA 12.x)	26 de abril de 2024	26 de abril de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-16:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-16:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-16:latest`
2,16	Cloud TPU	26 de abril de 2024	26 de abril de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-16:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-16:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-16:latest`
2.15	Somente CPU	15 de agosto de 2024	15 de agosto de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-15:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-15:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-15:latest`
2.15	GPU (CUDA 12.x)	15 de agosto de 2024	15 de agosto de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-15:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-15:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-15:latest`
2.15	Cloud TPU	15 de agosto de 2024	15 de agosto de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-15:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-15:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-15:latest`
2.14	Somente CPU	15 de agosto de 2024	15 de agosto de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-14:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-14:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-14:latest`
2.14	GPU (CUDA 12.x)	15 de agosto de 2024	15 de agosto de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-14:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-14:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-14:latest`
2,13	Somente CPU	15 de agosto de 2024	15 de agosto de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-13:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-13:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-13:latest`
2,13	GPU (CUDA 11.x)	15 de agosto de 2024	15 de agosto de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-13:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-13:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-13:latest`
2,12	Somente CPU	15 de maio de 2024	15 de maio de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-12:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-12:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-12:latest`
2,12	GPU (CUDA 11.x)	15 de maio de 2024	15 de maio de 2025	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-12:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-12:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-12:latest`
2.11	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-11:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-11:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-11:latest`
2.11	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-11:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-11:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-11:latest`
2.10	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-10:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-10:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-10:latest`
2.10	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-10:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-10:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-10:latest`
2.9	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-9:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-9:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-9:latest`
2.9	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-9:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-9:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-9:latest`
2.8	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-8:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-8:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-8:latest`
2.8	GPU (CUDA 11.x)	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-8:latest` `europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-8:latest` `asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-8:latest`

PyTorch

Versão do framework de MLs	Aceleradores compatíveis (e a versão CUDA, se aplicável)	Fim do patch e da data de suporte	Fim da disponibilidade	Imagens compatíveis
2.1 (Python 3.9)	Somente CPU	1º de dezembro de 2024	1º de dezembro de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-1:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-1:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-1:latest`
2.1 (Python 3.9)	GPU (CUDA 12.x)	1º de dezembro de 2024	1º de dezembro de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-1:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-1:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-1:latest`
2.1 (Python 3.9)	Cloud TPU	1º de dezembro de 2024	1º de dezembro de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-1:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-1:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-1:latest`
2.0 (Python 3.9)	Somente CPU	27 de julho de 2024	27 de julho de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-0:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-0:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-0:latest`
2.0 (Python 3.9)	GPU (CUDA 11.x)	27 de julho de 2024	27 de julho de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-0:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-0:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-0:latest`
1.13 (Python 3.8)	Somente CPU	15 de maio de 2024	15 de maio de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-13:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-13:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-13:latest`
1.13 (Python 3.8)	GPU (CUDA 11.x)	15 de maio de 2024	15 de maio de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-13:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-13:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-13:latest`
1.12	Somente CPU	15 de maio de 2024	15 de maio de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-12:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-12:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-12:latest`
1.12	GPU (CUDA 11.x)	15 de maio de 2024	15 de maio de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-12:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-12:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-12:latest`
1.11	Somente CPU	15 de maio de 2024	15 de maio de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest`
1.11	GPU (CUDA 11.x)	15 de maio de 2024	15 de maio de 2025	`us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-11:latest` `europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-11:latest` `asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-11:latest`

scikit-learn

Versão do framework de MLs	Aceleradores compatíveis (e a versão CUDA, se aplicável)	Fim do patch e da data de suporte	Fim da disponibilidade	Imagens compatíveis
1.3 (Python 3.10)	Somente CPU	28 de novembro de 2024	28 de novembro de 2025	`us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-3:latest` `europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-3:latest` `asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-3:latest`
1.2 (Python 3.10)	Somente CPU	30 de junho de 2024	30 de junho de 2025	`us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-2:latest` `europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-2:latest` `asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-2:latest`
1,0	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-0:latest` `europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-0:latest` `asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-0:latest`
0,24	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-24:latest` `europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-24:latest` `asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-24:latest`
0.23	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-23:latest` `europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-23:latest` `asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-23:latest`
0,22	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-22:latest` `europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-22:latest` `asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-22:latest`
0,20	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-20:latest` `europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-20:latest` `asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-20:latest`

XGBoost

Versão do framework de MLs	Aceleradores compatíveis (e a versão CUDA, se aplicável)	Fim do patch e da data de suporte	Fim da disponibilidade	Imagens compatíveis
1.7 (Python 3.10)	Somente CPU	30 de junho de 2024	30 de junho de 2025	`us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest` `europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest` `asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest`
1.6	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-6:latest` `europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-6:latest` `asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-6:latest`
1.5	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-5:latest` `europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-5:latest` `asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-5:latest`
1.4	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-4:latest` `europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-4:latest` `asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-4:latest`
1.3	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-3:latest` `europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-3:latest` `asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-3:latest`
1.2	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-2:latest` `europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-2:latest` `asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-2:latest`
1.1	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-1:latest` `europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-1:latest` `asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-1:latest`
0,90	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-90:latest` `europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-90:latest` `asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-90:latest`
0,82	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-82:latest` `europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-82:latest` `asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-82:latest`

Versões compatíveis do Ray na Vertex AI

Há imagens de contêiner do Ray na Vertex AI disponíveis. As seguintes versões de contêineres pré-criados são compatíveis:

Versão do Ray Aceleradores compatíveis (e a versão CUDA, se aplicável) Fim do patch e da data de suporte Fim da disponibilidade Imagens compatíveis

2.33.0 (Python 3.10)

Somente CPU

16 de setembro de 2025

16 de setembro de 2026

us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-33.py310:latest
europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-33.py310:latest
asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-33.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
absl-py 2.1.0 cloud-tpu-client 0.10 cloudml-hypertune python-json-logger 2.0.7 google-cloud-resource-manager 1.12.4 setuptools 69.5.1 kubernetes Outro Dependências do Deep Learning Containers	"ca-certificates-java" libatlas-base-dev liblapack-dev g++ libio-all-perl libyaml-0-2 Outro Dependências de Deep Learning Containers

2.33.0 (Python 3.10)

GPU (CUDA 11.x)

16 de setembro de 2025

16 de setembro de 2026

us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-33.py310:latest
europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-33.py310:latest
asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-33.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
absl-py 2.1.0 cloud-tpu-client 0.10 cloudml-hypertune python-json-logger 2.0.7 google-cloud-resource-manager 1.12.4 setuptools 69.5.1 kubernetes Outro Dependências do Deep Learning Containers	"ca-certificates-java" libatlas-base-dev liblapack-dev g++ libio-all-perl libyaml-0-2 Outro Dependências de Deep Learning Containers

2.9.3 (Python 3.10)

Somente CPU

7 de maio de 2025

7 de maio de 2026

us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-9.py310:latest
europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-9.py310:latest
asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-9.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
absl-py 2.0.0 cloud-tpu-client 0.10 cloudml-hypertune python-json-logger 2.0.7 google-cloud-resource-manager 1.11.0 Outro Dependências de Deep Learning Containers	"ca-certificates-java" libatlas-base-dev liblapack-dev g++ libio-all-perl libyaml-0-2 Outro Dependências de Deep Learning Containers

2.9.3 (Python 3.10)

GPU (CUDA 11.x)

7 de maio de 2025

7 de maio de 2026

us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-9.py310:latest
europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-9.py310:latest
asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-9.py310:latest

Dependências incluídas

Pacotes PyPI	Pacotes Ubuntu
absl-py 2.0.0 cloud-tpu-client 0.10 cloudml-hypertune python-json-logger 2.0.7 google-cloud-resource-manager 1.11.0 Outro Dependências de Deep Learning Containers	"ca-certificates-java" libatlas-base-dev liblapack-dev g++ libio-all-perl libyaml-0-2 Outro Dependências de Deep Learning Containers

Frameworks compatíveis com pipelines

O Vertex AI Pipelines é compatível com as seguintes versões do SDK do framework de ML:

SDK do Kubeflow Pipelines (KFP)
SDK do TensorFlow Extended (TFX)
Componentes de Pipeline do Google Cloud (GCPC)

Como esses frameworks são atualizados periodicamente para corrigir bugs e vulnerabilidades, é recomendável sempre usar a versão mais recente do patch e permanecer até o final da data de suporte. O suporte ao cliente não será fornecido se uma versão estiver além da data de suporte. Para mais informações, consulte a política de suporte do framework da Vertex AI.

Em geral, siga estas diretrizes ao usar modelos de pipeline:

Fique por dentro da versão de patch mais recente do framework de ML.
Faça upgrade dos modelos de pipeline novos e atuais para a versão secundária mais recente do framework de ML pelo menos uma vez a cada 12 meses.
Quando uma nova versão principal de um framework de ML for disponibilizada, migre modelos de pipeline novos e existentes para a nova versão principal em até 24 meses.

SDK do Kubeflow Pipelines

O KFP é o framework mais usado para criar e orquestrar pipelines no Vertex AI Pipelines. Os Vertex AI Pipelines podem executar definições de pipeline compiladas usando o KFP na representação intermediária pipeline_spec (formato YAML IR) v2.0 ou posterior.

O Vertex AI Pipelines é compatível com as seguintes versões do SDK do KFP:

Framework	Data de lançamento	Data de término do suporte
SDK do KFP 2	20 de junho de 2023	Não aplicável
KFP SDK 1.8	3 de setembro de 2021	20 de dezembro de 2024

A versão mais recente do SDK do KFP não vai dar suporte às versões do Python com término de vida útil no dia em que isso for declarado. A única exceção é o Python 3.7, que terá o suporte encerrado em 23 de abril de 2024.

SDK do TensorFlow Extended (TFX)

O TensorFlow lança uma nova versão secundária do SDK do TensorFlow Extended (TFX) a cada trimestre. Ainda é possível executar pipelines compilados em versões mais antigas do SDK, mas apenas a versão secundária mais recente do SDK recebe patches. Portanto, recomendamos que você se mantenha atualizado com a versão secundária mais recente.

Framework	Data de lançamento	Data de término do suporte
TensorFlow Extended SDK 1.15	29 de abril de 2024	29 de abril de 2025
TensorFlow Extended SDK 1.14	6 de setembro de 2023	6 de setembro de 2024
TensorFlow Extended SDK 1.13	3 de maio de 2023	3 de maio de 2024
TensorFlow Extended SDK 1.12	19 de dezembro de 2022	19 de dezembro de 2023
TensorFlow Extended SDK 1.11	23 de novembro de 2022	23 de novembro de 2023
TensorFlow Extended SDK 1.10	4 de outubro de 2022	4 de outubro de 2023
SDK estendido do TensorFlow 1.9	15 de julho de 2022	15 de julho de 2023
TensorFlow Extended SDK 1.8	25 de maio de 2022	25 de maio de 2023

Componentes do Google Cloud Pipeline

O Vertex AI Pipelines publica e corrige regularmente os seguintes pacotes Python e imagens de contêiner para ajudar nas definições de KFP:

Framework	Data de lançamento	Fim do patch e da data de suporte	Data de término da disponibilidade	Imagens compatíveis
Componentes 2 do Google Cloud Pipeline	20 de junho de 2023	Não aplicável	Não aplicável	`gcr.io/ml-pipeline/google-cloud-pipeline-components:latest`
Componentes 1 do Google Cloud Pipeline	25 de fevereiro de 2022	20 de dezembro de 2024	20 de junho de 2025	`gcr.io/ml-pipeline/google-cloud-pipeline-components:1.0.*`

A versão mais recente do SDK de componentes de pipeline do Google Cloud não vai dar mais suporte às versões do Python com término de vida útil no dia em que isso for declarado. A única exceção é o Python 3.7, que terá o suporte encerrado em 23 de abril de 2024.

Frameworks compatíveis com notebooks gerenciados pelo usuário do Vertex AI Workbench

Os notebooks gerenciados pelo usuário do Vertex AI Workbench são compatíveis com as versões de framework de ML nas tabelas a seguir. Para encontrar um VERSION_DATE específico para uma imagem, consulte Como listar as versões disponíveis.

Versões de base

Versão do framework de MLs	Versão atual do patch	Aceleradores compatíveis	Fim do patch e da data de suporte	Data de término da disponibilidade	Nome da família da imagem
CPU base (Python 3.10 / Debian 11)	Não relevante (N/A)	Somente CPU	1 de julho de 2024	1 de julho de 2025	`common-cpu-VERSION_DATE-debian-11`
Base-cu121 (Python 3.10)	CUDA 12.1	GPU (CUDA 12.1)	28 de fevereiro de 2024	28 de fevereiro de 2025	`common-cu121-VERSION_DATE-debian-11-py310`
Base-cu118 (Python 3.10)	CUDA 11.8	GPU (CUDA 11.8)	1 de julho de 2024	1 de julho de 2025	`common-cu118-VERSION_DATE-debian-11-py310`
Base-cu113 (Python 3.10)	CUDA 11.3	GPU (CUDA 11.3)	1 de janeiro de 2024	1 de janeiro de 2025	`common-cu113-VERSION_DATE-debian-11-py310`
Base-cu113 (Python 3.7)	CUDA 11.3	GPU (CUDA 11.3)	18 de setembro de 2023	18 de setembro de 2024	`common-cu113-VERSION_DATE-py37`
Base-cu110 (Python 3.7)	CUDA 11.0	GPU (CUDA 11.0)	18 de setembro de 2023	18 de setembro de 2024	`common-cu110-VERSION_DATE-py37`
CPU base (Python 3.7)	Não relevante (N/A)	Somente CPU	18 de setembro de 2023	18 de setembro de 2024	`common-cpu-VERSION_DATE-debian-10`

Versões do TensorFlow

Versão do framework de MLs	Versão atual do patch	Aceleradores compatíveis	Fim do patch e da data de suporte	Data de término da disponibilidade	Nome da família da imagem
2.15 (Python 3.10)	2.15.0	Somente CPU	14 de novembro de 2024	14 de novembro de 2024	`tf-2-15-cpu-VERSION_DATE-py310`
2.15 (Python 3.10)	2.15.0	GPU (CUDA 12.1)	14 de novembro de 2024	14 de novembro de 2024	`tf-2-15-cu121-VERSION_DATE-py310`
2.14 (Python 3.10)	2.14.0	Somente CPU	26 de setembro de 2024	26 de setembro de 2025	`tf-2-14-cpu-VERSION_DATE-py310`
2.14 (Python 3.10)	2.14.0	GPU (CUDA 11.8)	26 de setembro de 2024	26 de setembro de 2025	`tf-2-14-cu118-VERSION_DATE-py310`
2.13 (Python 3.10)	2.13.0	Somente CPU	5 de julho de 2024	5 de julho de 2025	`tf-2-13-cpu-VERSION_DATE-py310`
2.13 (Python 3.10)	2.13.0	GPU (CUDA 11.8)	5 de julho de 2024	5 de julho de 2025	`tf-2-13-cu118-VERSION_DATE-py310`
2.12 (Python 3.10)	2.12.0	Somente CPU	30 de junho de 2024	30 de junho de 2025	`tf-2-12-cpu-VERSION_DATE-py310`
2.12 (Python 3.10)	2.12.0	GPU (CUDA 11.8)	18 de janeiro de 2024	18 de janeiro de 2025	`tf-2-12-cu113-VERSION_DATE-py310`
2.11 (Python 3.10)	2.11.0	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`tf-2-11-cpu-VERSION_DATE-py310`
2.11 (Python 3.10)	2.11.0	GPU (CUDA 11.3)	15 de novembro de 2023	15 de novembro de 2024	`tf-2-11-cu113-VERSION_DATE-py310`
2.11	2.11.0	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`tf-2-11-cpu-VERSION_DATE-py37`
2.11	2.11.0	GPU (CUDA 11.3)	15 de novembro de 2023	15 de novembro de 2024	`tf-2-11-cu113-VERSION_DATE-py37`
2.10	2.10.1	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`tf-2-10-cpu-VERSION_DATE-py37`
2.10	2.10.1	GPU (CUDA 11.3)	15 de novembro de 2023	15 de novembro de 2024	`tf-2-10-cu113-VERSION_DATE-py37`
2.9	2.9.3	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`tf-2-9-cpu-VERSION_DATE-py37`
2.9	2.9.3	GPU (CUDA 11.3)	15 de novembro de 2023	15 de novembro de 2024	`tf-2-9-cu113-VERSION_DATE-py37`
2.8	2.8.4	Somente CPU	15 de novembro de 2023	15 de novembro de 2024	`tf-2-8-cpu-VERSION_DATE-py37`
2.8	2.8.4	GPU (CUDA 11.3)	15 de novembro de 2023	15 de novembro de 2024	`tf-2-8-cu113-VERSION_DATE-py37`
2.6 (py39)	2.6.5	Somente CPU	1 de setembro de 2023	1 de setembro de 2024	`tf-2-6-cpu-VERSION_DATE-py39`
2.6 (py39)	2.6.5	GPU (CUDA 11.3)	1 de setembro de 2023	1 de setembro de 2024	`tf-2-6-cu110-VERSION_DATE-py39`
2.6 (py37)	2.6.5	Somente CPU	18 de setembro de 2023	18 de setembro de 2024	`tf-2-6-cpu-VERSION_DATE-py37`
2.6 (py37)	2.6.5	GPU (CUDA 11.3)	18 de setembro de 2023	18 de setembro de 2024	`tf-2-6-cu110-VERSION_DATE-py37`
2.3	2.3.4	Somente CPU	1 de setembro de 2023	1 de setembro de 2024	`tf-2-3-cpu`
2.3	2.3.4	GPU (CUDA 11.3)	1 de setembro de 2023	1 de setembro de 2024	`tf-2-3-cu110-VERSION_DATE`

Versões do PyTorch

Versão do framework de MLs	Versão atual do patch	Aceleradores compatíveis	Fim do patch e da data de suporte	Data de término da disponibilidade	Nome da família da imagem
2.2 (Python 3.10)	2.2.0	CUDA 12.1	30 de janeiro de 2024	30 de janeiro de 2025	`pytorch-2-2-VERSION_DATE-py310`
2.1 (Python 3.10)	2.1.0	CUDA 12.1	4 de outubro de 2024	4 de outubro de 2025	`pytorch-2-1-VERSION_DATE-py310`
2.0 (Python 3.10)	2.0.0	CUDA 11.8	15 de março de 2024	15 de março de 2025	`pytorch-2-0-VERSION_DATE-py310`
1.13 (Python 3.10)	1.13.1	CUDA 11.3	8 de dezembro de 2023	8 de dezembro de 2024	`pytorch-1-13-VERSION_DATE-py310`
1.13	1.13.1	CUDA 11.3	8 de dezembro de 2023	8 de dezembro de 2024	`pytorch-1-13-VERSION_DATE-py37`
1.12	1.12.1	CUDA 11.3	1 de setembro de 2023	1 de setembro de 2024	`pytorch-1-12-VERSION_DATE-py310`

A seguir

Saiba mais sobre a responsabilidade compartilhada da Vertex AI.