Contêineres pré-criados para treinamento personalizado

A Vertex AI fornece imagens de contêiner do Docker que você executa como contêineres pré-criados para treinamento personalizado. Esses contêineres, organizados por framework de machine learning (ML) e versão do framework, incluem dependências comuns que você pode querer usar no código de treinamento. Muitas vezes, usar um contêiner pré-criado é mais simples do que criar seu próprio contêiner personalizado para treinamento.

Neste documento, listamos os contêineres pré-criados para treinamento e descrevemos como usá-los com um aplicativo de treinamento em Python.

A Vertex AI é compatível com cada versão de framework com base em uma programação para minimizar as vulnerabilidades de segurança. Consulte a política de suporte do framework da Vertex AI para entender as implicações das datas de término e de fim da disponibilidade do suporte.

Imagens de contêiner disponíveis

Cada uma das imagens de contêiner a seguir está disponível em vários repositórios do Artifact Registry, que armazenam dados em vários locais. É possível usar qualquer um dos URIs de uma imagem ao realizar o treinamento personalizado. cada um deles fornece a mesma imagem de contêiner. Ao usar o console do Google Cloud para realizar um treinamento personalizado, o Console do Google Cloud seleciona o URI que melhor corresponde ao local em que você está usando a Vertex AI. Isso reduz a latência.

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 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.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 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.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 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 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.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 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.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 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 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.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 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.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 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 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.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 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.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 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 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.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 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.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 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 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.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 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.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 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 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
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 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
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 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 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

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.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 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 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 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 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 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
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 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
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 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
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 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

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 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.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 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 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

Treinar com contêineres pré-criados

Para ver um exemplo de uso de um contêiner pré-criado para treinamento como parte de um fluxo de trabalho mais abrangente, execute o notebook Jupyter "Treinamento personalizado e previsão on-line" em um dos seguintes ambientes:

Abrir no Colab | Abrir no Colab Enterprise | Abrir nos notebooks gerenciados pelo usuário do Vertex AI Workbench | Conferir no GitHub

Para usar um contêiner pré-criado, leia o guia de como definir configurações de contêiner.

Se você estiver usando uma imagem de contêiner compatível com GPUs, especifique os campos machineSpec.acceleratorType e machineSpec.acceleratorCount ao criar seu recurso CustomJob, HyperparameterTuningJob ou TrainingPipeline.

A seguir

• Saiba como criar um aplicativo de treinamento em Python para usar com um contêiner pré-criado.