Lista de 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 Vertex AI.

Frameworks compatíveis com o treinamento personalizado

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

TensorFlow

Versão do framework de MLs Aceleradores compatíveis (e a versão CUDA, se aplicável) Fim do patch e da data de suporte Fim da disponibilidade Imagens compatíveis
2.15 (Python 3.10) Somente TPU 23 de outubro de 2024 23 de outubro de 2025
  • us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
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.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.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.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
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 Vertex AI usa imagens de contêiner para configurar recursos de nuvem para atender às solicitações de previsão. As versões a seguir são compatíveis com contêineres pré-criados para previsões e explicações:

TensorFlow

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

Ambiente de execução otimizado do TensorFlow

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

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

PyTorch

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

scikit-learn

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

XGBoost

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

Versões compatíveis do Ray na Vertex AI

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

Versão do Ray Aceleradores compatíveis (e a versão CUDA, se aplicável) Fim do patch e da data de suporte Fim da disponibilidade Imagens compatíveis
2.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 Vertex AI 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 Vertex AI.

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

  • Fique por dentro da versão de patch mais recente do framework de ML.

  • Faça upgrade dos modelos de pipeline novos e atuais para a versão secundária mais recente do framework de ML pelo menos uma vez a cada 12 meses.

  • Quando uma nova versão principal de um framework de ML for disponibilizada, migre modelos de pipeline novos e existentes para a nova versão principal em até 24 meses.

SDK do Kubeflow Pipelines

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

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

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

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

SDK do TensorFlow Extended (TFX)

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

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

Componentes do Google Cloud Pipeline

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

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

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

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

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

Versões de base

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

Versões do TensorFlow

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

Versões do PyTorch

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

A seguir