Frameworks compatíveis

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

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

Frameworks compatíveis com o treinamento personalizado

A plataforma de agentes do Gemini Enterprise usa imagens de contêiner para configurar recursos de nuvem e 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.17 (Python 3.10) Somente 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) Somente 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) Somente CPU 14 de novembro de 2024 14 de novembro de 2024
  • 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 2024
  • 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) 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
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
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) 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
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) 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
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) 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
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
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) 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
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) 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
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) 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
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) 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
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) 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
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) 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
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) 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
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) 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
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) 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
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) 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
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) 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
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) 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
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) 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
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 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.4 (Python 3.10) Somente 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) Somente 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) 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
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
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
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
2.2 (Python 3.10) Somente 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) 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
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
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
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
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
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
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 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
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
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
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
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) 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
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) 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
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) 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
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) 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
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) 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
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 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 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) 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
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) 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

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
2.1 (Python 3.10) Somente 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) 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
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) 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

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

A plataforma de agentes do Gemini Enterprise usa imagens de contêiner para configurar recursos de nuvem e 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

Imagens de contêiner do TensorFlow disponíveis (clique para expandir)

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 Somente CPU 14 de julho de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-15:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-15:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-15:latest
2.15 GPU (CUDA 12.x) 14 de julho de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-15:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-15:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-15:latest
2.14 Somente CPU 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-14:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-14:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-14:latest
2.14 GPU (CUDA 12.x) 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-14:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-14:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-14:latest
2,13 Somente CPU 28 de novembro de 2024 28 de novembro de 2025
2,13 GPU (CUDA 12.x) 28 de novembro de 2024 28 de novembro de 2025
2,12 Somente CPU 30 de junho de 2024 30 de junho de 2025
2,12 GPU (CUDA 11.x) 30 de junho de 2024 30 de junho de 2025
2.11 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.11 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.10 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.10 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.9 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.9 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.8 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.8 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.7 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.7 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2,6 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2,6 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.5 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.5 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2,4 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2,4 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.3 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.3 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.2 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.2 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.1 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.1 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
1.15 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
1.15 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024

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.

Imagens de contêiner do ambiente de execução otimizado do TensorFlow disponíveis (clique para abrir)

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 relevante
  • 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,17 Somente CPU 11 de julho de 2024 11 de julho de 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-17:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-17:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-17:latest
2,17 GPU (CUDA 12.x) 11 de julho de 2024 11 de julho de 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-17:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-17:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-17:latest
2,17 Cloud TPU 11 de julho de 2024 11 de julho de 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-17:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-17:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-17:latest
2,16 Somente CPU 26 de abril de 2024 26 de abril de 2025
2,16 GPU (CUDA 12.x) 26 de abril de 2024 26 de abril de 2025
2,16 Cloud TPU 26 de abril de 2024 26 de abril de 2025
2.15 Somente CPU 15 de agosto de 2024 15 de agosto de 2025
2.15 GPU (CUDA 12.x) 15 de agosto de 2024 15 de agosto de 2025
2.15 Cloud TPU 15 de agosto de 2024 15 de agosto de 2025
2.14 Somente CPU 15 de agosto de 2024 15 de agosto de 2025
2.14 GPU (CUDA 12.x) 15 de agosto de 2024 15 de agosto de 2025
2,13 Somente CPU 15 de agosto de 2024 15 de agosto de 2025
2,13 GPU (CUDA 11.x) 15 de agosto de 2024 15 de agosto de 2025
2,12 Somente CPU 15 de maio de 2024 15 de maio de 2025
2,12 GPU (CUDA 11.x) 15 de maio de 2024 15 de maio de 2025
2.11 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.11 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.10 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.10 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.9 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.9 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024
2.8 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
2.8 GPU (CUDA 11.x) 15 de novembro de 2023 15 de novembro de 2024

PyTorch

Imagens de contêiner do PyTorch disponíveis (clique para expandir)

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.4 (Python 3.9) Somente CPU 14 de julho de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-4:latest
2.4 (Python 3.9) GPU (CUDA 12.x) 14 de julho de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-4:latest
2.4 (Python 3.9) Cloud TPU 14 de julho de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-4:latest
2.3 (Python 3.9) Somente CPU 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-3:latest
2.3 (Python 3.9) GPU (CUDA 12.x) 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-3:latest
2.3 (Python 3.9) Cloud TPU 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-3:latest
2.2 (Python 3.9) Somente CPU 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-2:latest
2.2 (Python 3.9) GPU (CUDA 12.x) 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-2:latest
2.2 (Python 3.9) Cloud TPU 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-2:latest
2.1 (Python 3.9) Somente CPU 1º de dezembro de 2024 1º de dezembro de 2025
2.1 (Python 3.9) GPU (CUDA 12.x) 1º de dezembro de 2024 1º de dezembro de 2025
2.1 (Python 3.9) Cloud TPU 1º de dezembro de 2024 1º de dezembro de 2025
2.0 (Python 3.9) Somente CPU 27 de julho de 2024 27 de julho de 2025
2.0 (Python 3.9) GPU (CUDA 11.x) 27 de julho de 2024 27 de julho de 2025
1.13 (Python 3.8) Somente CPU 15 de maio de 2024 15 de maio de 2025
1.13 (Python 3.8) GPU (CUDA 11.x) 15 de maio de 2024 15 de maio de 2025
1.12 Somente CPU 15 de maio de 2024 15 de maio de 2025
1.12 GPU (CUDA 11.x) 15 de maio de 2024 15 de maio de 2025
1.11 Somente CPU 15 de maio de 2024 15 de maio de 2025
1.11 GPU (CUDA 11.x) 15 de maio de 2024 15 de maio de 2025

scikit-learn

Imagens de contêiner do scikit-learn disponíveis (clique para abrir)

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 14 de outubro de 2026 14 de outubro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-6:latest
1.5 (Python 3.10) Somente CPU 14 de julho de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-5:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-5:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-5:latest
1.4 (Python 3.10) Somente CPU 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-4:latest
1.3 (Python 3.10) Somente CPU 28 de novembro de 2024 28 de novembro de 2025
1.2 (Python 3.10) Somente CPU 30 de junho de 2024 30 de junho de 2025
1,0 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
0,24 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
0.23 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
0,22 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
0,20 Somente CPU 15 de novembro de 2023 15 de novembro de 2024

XGBoost

Imagens de contêiner XGBoost disponíveis (clique para abrir)

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.10) Somente CPU 14 de julho de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-1:latest
2.0 (Python 3.10) Somente CPU 14 de janeiro de 2026 14 de janeiro de 2027
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-0:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-0:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.2-0:latest
1.7 (Python 3.10) Somente CPU 30 de junho de 2024 30 de dezembro de 2025
1.6 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
1.5 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
1.4 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
1.3 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
1.2 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
1.1 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
0,90 Somente CPU 15 de novembro de 2023 15 de novembro de 2024
0,82 Somente CPU 15 de novembro de 2023 15 de novembro de 2024

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.47.1 (Python 3.11) Somente 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) Somente 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) Somente 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) 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
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) 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
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

Frameworks compatíveis com pipelines

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

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 plataforma de agentes do Gemini Enterprise.

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 com frequência para criar e orquestrar pipelines no Agent Platform Pipelines. Os pipelines da Agent Platform 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 Agent Platform 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

Google Cloud Componentes de pipeline

O Agent Platform 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
Google Cloud Componentes do pipeline 2 20 de junho de 2023 Não aplicável Não relevante gcr.io/ml-pipeline/google-cloud-pipeline-components:latest
Google Cloud Componentes do pipeline 1 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 Google Cloud não vai mais 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.

A seguir