Contentores pré-criados para preparação personalizada

O Vertex AI fornece imagens de contentores Docker que executa como contentores pré-criados para a preparação personalizada. Estes contentores, que estão organizados por framework de aprendizagem automática (AA) e versão da framework, incluem dependências comuns que pode querer usar no código de preparação. Muitas vezes, usar um contentor pré-criado é mais simples do que criar o seu próprio contentor personalizado para formação.

Este documento lista os contentores pré-criados para preparação e descreve como os usar com uma aplicação de preparação Python.

O Vertex AI suporta cada versão do framework com base num agendamento para minimizar as vulnerabilidades de segurança. Reveja a Política de apoio técnico de frameworks do Vertex AI para compreender as implicações das datas de fim do apoio técnico e de fim da disponibilidade.

Imagens de contentores disponíveis

Cada uma das seguintes imagens de contentores está disponível em vários repositórios do Artifact Registry, que armazenam dados em várias localizações. Pode usar qualquer um dos URIs para uma imagem quando realizar uma preparação personalizada. Cada um fornece a mesma imagem do contentor. Quando usa a Google Cloud consola para realizar a preparação personalizada, a Google Cloud consola seleciona o URI que melhor corresponde à localização onde está a usar o Vertex AI. Isto reduz a latência.

TensorFlow

Versão do framework de ML Aceleradores suportados (e versão CUDA, se aplicável) Data de fim da aplicação de patches e do apoio técnico Fim da disponibilidade Imagens suportadas
2.17 (Python 3.10) Apenas CPU 11 de julho de 2025 11 de julho de 2026
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
2.17 (Python 3.10) GPU (CUDA 12.x) 11 de julho de 2025 11 de julho de 2026
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
2.16 (Python 3.10) Apenas CPU 28 de junho de 2025 28 de junho de 2026
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
2.16 (Python 3.10) GPU (CUDA 12.x) 28 de junho de 2025 28 de junho de 2026
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
2.15 (Python 3.10) Apenas CPU 14 de novembro de 2024 14 de novembro de 2025
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
2.15 (Python 3.10) GPU (CUDA 12.x) 14 de novembro de 2024 14 de novembro de 2025
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
2.15 (Python 3.10) Apenas TPU 23 de out. 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
2.14 (Python 3.10) Apenas 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
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
2.13 (Python 3.10) Apenas 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
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
2.12 (Python 3.10) Apenas 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
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
2.12 (Python 3.8) Apenas 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
2.11 (Python 3.10) Apenas 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
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
2.11 (Python 3.7) Apenas 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
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
2.9 (Python 3.10) Apenas 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
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
2.9 (Python 3.7) Apenas 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
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
2.8 (Python 3.10) Apenas 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
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
2.8 (Python 3.7) Apenas 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
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
2.7 (Python 3.7) Apenas 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
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
2.6 (Python 3.7) Apenas 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
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
2.5 (Python 3.7) Apenas 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
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
2.4 (Python 3.7) Apenas 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
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
2.3 (Python 3.7) Apenas 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
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
2.2 (Python 3.7) Apenas 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
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
2.1 (Python 3.7) Apenas 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
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
1.15 (Python 3.7) Apenas 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
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

PyTorch

Versão do framework de ML Aceleradores suportados (e versão CUDA, se aplicável) Data de fim da aplicação de patches e do apoio técnico Fim da disponibilidade Imagens suportadas
2.4 (Python 3.10) Apenas CPU 24 de julho de 2025 24 de julho de 2026
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-4.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-4.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-4.py310:latest
2.4 (Python 3.10) Apenas GPU (CUDA 12.x) 24 de julho de 2025 24 de julho de 2026
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest
2.3 (Python 3.10) Apenas 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
2.3 (Python 3.10) Apenas 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
2.2 (Python 3.10) Apenas CPU 30 de janeiro de 2025 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
2.2 (Python 3.10) Apenas GPU (CUDA 12.x) 30 de janeiro de 2025 30 de janeiro de 2026
  • 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
2.2 (Python 3.10) Apenas TPU 6 de novembro de 2024 6 de novembro de 2025
  • us-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-2.cp310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-2.cp310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-2.cp310:latest
2.1 (Python 3.10) Apenas 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
2.1 (Python 3.10) Apenas 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
2.1 (Python 3.8) Apenas TPU 23 de out. 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
2.0 (Python 3.10) Apenas 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
2.0 (Python 3.10) Apenas 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
2.0 (Python 3.8) Apenas 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
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
1.13 (Python 3.7) CPU ou GPU (CUDA 11.x) 8 de dezembro de 2023 8 de dez. 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
1.12 (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-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
1.12 (Python 3.7) CPU ou 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
1.11 (Python 3.7) Apenas 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
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
1.10 (Python 3.7) Apenas 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
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
1.9 (Python 3.7) Apenas 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
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
1.7 (Python 3.7) Apenas 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
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
1.6 (Python 3.7) Apenas 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
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
1.4 (Python 3.7) Apenas 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
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

scikit-learn

Versão do framework de ML Aceleradores suportados (e versão CUDA, se aplicável) Data de fim da aplicação de patches e do apoio técnico Fim da disponibilidade Imagens suportadas
1.6 (Python 3.10) Apenas CPU 11 de julho de 2025 11 de julho de 2026
  • us-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-6:latest
1.6 (Python 3.10) GPU (CUDA 12.x) 11 de julho de 2025 11 de julho de 2026
  • us-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-6:latest
1.0 (Python 3.10) Apenas 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
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
0,23 (Python 3.7) Apenas 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

XGBoost

Versão do framework de ML Aceleradores suportados (e versão CUDA, se aplicável) Data de fim da aplicação de patches e do apoio técnico Fim da disponibilidade Imagens suportadas
2.1 (Python 3.10) Apenas CPU 11 de julho de 2025 11 de julho de 2026
  • us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.2-1:latest
2.1 (Python 3.10) GPU (CUDA 12.x) 11 de julho de 2025 11 de julho de 2026
  • us-docker.pkg.dev/vertex-ai/training/xgboost-gpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-gpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-gpu.2-1:latest
1.6 (Python 3.10) Apenas 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
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
1.1 (Python 3.7) Apenas 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

Versões suportadas do Ray no Vertex AI

As imagens de contentores do Ray no Vertex AI estão disponíveis. As seguintes versões de contentores pré-criados são suportadas:

Versão do Ray Aceleradores suportados (e versão CUDA, se aplicável) Data de fim da aplicação de patches e do apoio técnico Fim da disponibilidade Imagens suportadas
2.47.1 (Python 3.11) Apenas CPU 28 de julho de 2026 28 de julho de 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-47.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-47.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-47.py311:latest
2.47.1 (Python 3.11) GPU (CUDA 11.x) 28 de julho de 2026 28 de julho de 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-47.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-47.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-47.py311:latest
2.42.0 (Python 3.11) Apenas CPU 1 de março de 2026 1 de março de 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py311:latest
2.42.0 (Python 3.11) GPU (CUDA 11.x) 1 de março de 2026 1 de março de 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py311:latest
2.42.0 (Python 3.10) Apenas CPU 1 de março de 2026 1 de março de 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py310:latest
2.42.0 (Python 3.10) GPU (CUDA 11.x) 1 de março de 2026 1 de março de 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py310:latest
2.33.0 (Python 3.10) Apenas 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
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
2.9.3 (Python 3.10) Apenas 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
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

Faça a preparação com um contentor pré-criado

Para usar um contentor pré-criado, leia o guia de configuração das definições do contentor.

Se estiver a usar uma imagem de contentor que suporte GPUs, certifique-se de que especifica os campos machineSpec.acceleratorType e machineSpec.acceleratorCount quando criar o recurso CustomJob, HyperparameterTuningJob ou TrainingPipeline.

O que se segue?