Liste des frameworks compatibles

Cette page répertorie les frameworks compatibles et leurs packages constitutifs pour les services suivants :

Pour une explication détaillée des règles de compatibilité du framework, consultez la page Règles de compatibilité du framework Vertex AI.

Frameworks compatibles avec l'entraînement personnalisé

Vertex AI utilise des images de conteneurs pour configurer des ressources cloud afin de répondre à vos requêtes d'entraînement. Les versions suivantes sont compatibles avec les conteneurs prédéfinis d'entraînement personnalisé :

TensorFlow

Version du framework de ML Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
2.15 (Python 3.10) TPU uniquement 23 octobre 2023 23 octobre 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) CPU uniquement 26 septembre 2024 26 septembre 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 septembre 2024 26 septembre 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) CPU uniquement 5 juillet 2024 5 juillet 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 juillet 2024 5 juillet 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) CPU uniquement 30 juin 2024 30 juin 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 juin 2024 30 juin 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
Python TPU uniquement 30 juin 2024 30 juin 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) CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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) CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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) CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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) CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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) CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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) CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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

Version du framework de ML Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
2.3 (Python 3.10) CPU uniquement 24 avril 2025 24 avril 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) GPU uniquement (CUDA 12.x) 24 avril 2025 24 avril 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) CPU uniquement 30 janvier 2025 23 octobre 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) GPU uniquement (CUDA 12.x) 30 janvier 2025 30 janvier 2026
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest
2.1 (Python 3.10) CPU uniquement 4 octobre 2024 4 octobre 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) GPU uniquement (CUDA 12.x) 4 octobre 2024 4 octobre 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) TPU uniquement 23 octobre 2023 23 octobre 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) CPU uniquement 15 mars 2024 15 mars 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) GPU uniquement (CUDA 11.x) 15 mars 2024 15 mars 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) TPU uniquement 15 mai 2024 15 mai 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 mai 2024 15 mai 2025
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest
1.13 (Python 3.7) CPU ou GPU (CUDA 11.x) 8 décembre 2023 8 décembre 2024
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest
1.12 (Python 3.10) CPU ou GPU (CUDA 11.x) 15 mai 2024 15 mai 2025
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest
1.12 (Python 3.7) CPU ou GPU (CUDA 11.x) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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) CPU uniquement 1er septembre 2023 1er septembre 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) 1er septembre 2023 1er septembre 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

Version du framework de ML Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
1.0 (Python 3.10) CPU uniquement 30 juin 2024 30 juin 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 juin 2024 30 juin 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) CPU uniquement 1er septembre 2023 1er septembre 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

Version du framework de ML Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
1.6 (Python 3.10) CPU uniquement 30 juin 2024 30 juin 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 juin 2024 30 juin 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) CPU uniquement 15 novembre 2023 15 novembre 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 compatibles avec les prédictions et les explications

Vertex AI utilise des images de conteneurs pour configurer des ressources cloud afin de répondre à vos requêtes de prédiction. Les versions suivantes sont compatibles avec les conteneurs prédéfinis pour les prédictions et les explications :

TensorFlow

Version du framework de ML Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
2,13 CPU uniquement 28 novembre 2024 28 novembre 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 novembre 2024 28 novembre 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 CPU uniquement 30 juin 2024 30 juin 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 juin 2024 30 juin 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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

Environnement d'exécution TensorFlow optimisé

Les images de conteneur suivantes utilisent l'environnement d'exécution TensorFlow optimisé. Pour en savoir plus, consultez la page Utiliser l'environnement d'exécution TensorFlow optimisé.

Version du framework de ML Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
nightly CPU uniquement Non applicable Non applicable
  • 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
nightly GPU (CUDA 12.x) Non applicable Non applicable
  • 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
nightly Cloud TPU Non applicable Non applicable
  • 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 CPU uniquement 26 avril 2024 26 avril 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 avril 2024 26 avril 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 avril 2024 26 avril 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 CPU uniquement 15 août 2024 15 août 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 août 2024 15 août 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 août 2024 15 août 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 CPU uniquement 15 août 2024 15 août 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 août 2024 15 août 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 CPU uniquement 15 août 2024 15 août 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 août 2024 15 août 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 CPU uniquement 15 mai 2024 15 mai 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 mai 2024 15 mai 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 novembre 2023 15 novembre 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

Version du framework de ML Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
2.1 (Python 3.9) CPU uniquement 1er décembre 2024 1er décembre 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) 1er décembre 2024 1er décembre 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 1er décembre 2024 1er décembre 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) CPU uniquement 27 juillet 2024 27 juillet 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 juillet 2024 27 juillet 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) CPU uniquement 15 mai 2024 15 mai 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 mai 2024 15 mai 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 CPU uniquement 15 mai 2024 15 mai 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 mai 2024 15 mai 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 CPU uniquement 15 mai 2024 15 mai 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 mai 2024 15 mai 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

Version du framework de ML Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
1.3 (Python 3.10) CPU uniquement 28 novembre 2024 28 novembre 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) CPU uniquement 30 juin 2024 30 juin 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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

Version du framework de ML Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
1.7 (Python 3.10) CPU uniquement 30 juin 2024 30 juin 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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 CPU uniquement 15 novembre 2023 15 novembre 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

Versions compatibles de Ray sur Vertex AI

Les images de conteneurs Ray sur Vertex AI sont disponibles. Les versions suivantes des conteneurs prédéfinis sont compatibles:

Version Ray Accélérateurs compatibles (et version CUDA, le cas échéant) Date de fin de correctif et de fin de compatibilité Fin de disponibilité Images compatibles
2.33.0 (Python 3.10) CPU uniquement 16 septembre 2025 16 septembre 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 septembre 2025 16 septembre 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) CPU uniquement 7 mai 2025 7 mai 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 mai 2025 7 mai 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 compatibles avec les pipelines

Vertex AI Pipelines est compatible avec les versions de SDK de framework de ML suivantes :

Comme ces frameworks sont régulièrement mis à jour pour corriger les bugs et les failles, nous vous recommandons de toujours utiliser la dernière version du correctif et de ne pas dépasser la date de fin de prise en charge. Le service client n'est pas fourni si une version dépasse la date de fin de prise en charge. Pour en savoir plus, consultez les règles de compatibilité du framework Vertex AI.

En général, suivez les consignes suivantes lorsque vous utilisez des modèles de pipeline :

  • Veillez à vous maintenir à jour avec la dernière version de correctif du framework de ML.

  • Mettez à niveau les modèles de pipeline nouveaux et existants vers la dernière version mineure du framework de ML au moins une fois tous les 12 mois.

  • Lorsqu'une nouvelle version majeure d'un framework de ML est disponible, migrez les modèles de pipeline nouveaux et existants vers la nouvelle version majeure dans les 24 mois qui suivent.

SDK Kubeflow Pipelines

KFP est le framework le plus couramment utilisé pour créer et orchestrer des pipelines dans Vertex AI Pipelines. Vertex AI Pipelines peut exécuter des définitions de pipeline compilées à l'aide de KFP dans la représentation intermédiaire pipeline_spec (format IR YAML) version 2.0 ou ultérieure.

Vertex AI Pipelines est compatible avec les versions suivantes du SDK KFP :

Framework Date de sortie Date de fin d'assistance
SDK KFP 2 20 juin 2023 Non applicable
SDK KFP 1.8 3 septembre 2021 20 décembre 2024

La dernière version du SDK KFP ne sera plus compatible avec les versions de fin de vie de Python à compter du jour où elles sont déclarées comme arrivées en fin de vie. La seule exception à cette règle est Python 3.7, pour laquelle la compatibilité ne sera plus assurée à partir du 23 avril 2024.

SDK TensorFlow Extended (TFX)

TensorFlow publie une nouvelle version mineure du SDK TensorFlow Extended (TFX) tous les trimestres environ. Vous pouvez toujours exécuter des pipelines compilés à partir d'anciennes versions du SDK, mais seule la dernière version mineure du SDK reçoit des correctifs. Par conséquent, nous vous recommandons de vous tenir informé de la dernière version mineure.

Framework Date de sortie Date de fin d'assistance
SDK TensorFlow Extended 1.15 29 avril 2024 29 avril 2025
SDK TensorFlow Extended 1.14 6 septembre 2023 6 septembre 2024
SDK TensorFlow Extended 1.13 3 mai 2023 3 mai 2024
SDK TensorFlow Extended 1.12 19 décembre 2022 19 décembre 2023
SDK TensorFlow Extended 1.11 23 novembre 2022 23 novembre 2023
SDK TensorFlow Extended 1.10 4 octobre 2022 4 octobre 2023
SDK TensorFlow Extended 1.9 15 juillet 2022 15 juillet 2023
SDK TensorFlow Extended 1.8 25 mai 2022 25 mai 2023

Composants du pipeline Google Cloud

Vertex AI Pipelines publie et corrige régulièrement les packages Python et les images de conteneur suivants pour faciliter les définitions KFP :

Framework Date de sortie Date de fin de correctif et de fin de compatibilité Date de fin de disponibilité Images compatibles
Composants du pipeline Google Cloud 2 20 juin 2023 Non applicable Non applicable gcr.io/ml-pipeline/google-cloud-pipeline-components:latest
Composants du pipeline Google Cloud 1 25 février 2022 20 décembre 2024 20 juin 2025 gcr.io/ml-pipeline/google-cloud-pipeline-components:1.0.*

La dernière version du SDK des composants du pipeline Google Cloud ne sera plus compatible avec les versions de fin de vie de Python à compter du jour où elles sont déclarées comme arrivées en fin de vie. La seule exception à cette règle est Python 3.7, pour laquelle la compatibilité ne sera plus assurée à partir du 23 avril 2024.

Frameworks compatibles avec les notebooks gérés par l'utilisateur Vertex AI Workbench

Les notebooks gérés par l'utilisateur Vertex AI Workbench sont compatibles avec les versions de framework de ML des tableaux suivants. Pour trouver une valeur VERSION_DATE spécifique pour une image, consultez la section Répertorier les versions disponibles.

Versions de base

Version du framework de ML Version de correctif actuelle Accélérateurs compatibles Date de fin de correctif et de fin de compatibilité Date de fin de disponibilité Nom de la famille d'images
Processeur de base (Python 3.10 / Debian 11) Non applicable (N/A) CPU uniquement 1er juillet 2024 1er juillet 2025 common-cpu-VERSION_DATE-debian-11
Base-cu121 (Python 3.10) CUDA 12.1 GPU (CUDA 12.1) 28 février 2024 28 février 2025 common-cu121-VERSION_DATE-debian-11-py310
Base-cu118 (Python 3.10) CUDA 11.8 GPU (CUDA 11.8) 1er juillet 2024 1er juillet 2025 common-cu118-VERSION_DATE-debian-11-py310
Base-cu113 (Python 3.10) CUDA 11.3 GPU (CUDA 11.3) 1er janvier 2024 1er janvier 2025 common-cu113-VERSION_DATE-debian-11-py310
Base-cu113 (Python 3.7) CUDA 11.3 GPU (CUDA 11.3) 18 sept 2023 18 sept 2024 common-cu113-VERSION_DATE-py37
Base-cu110 (Python 3.7) CUDA 11.0 GPU (CUDA 11.0) 18 sept 2023 18 sept 2024 common-cu110-VERSION_DATE-py37
Processeur de base (Python 3.7) Non applicable (N/A) CPU uniquement 18 sept 2023 18 sept 2024 common-cpu-VERSION_DATE-debian-10

Versions de TensorFlow

Version du framework de ML Version de correctif actuelle Accélérateurs compatibles Date de fin de correctif et de fin de compatibilité Date de fin de disponibilité Nom de la famille d'images
2.15 (Python 3.10) 2.15.0 CPU uniquement 14 novembre 2024 14 novembre 2025 tf-2-15-cpu-VERSION_DATE-py310
2.15 (Python 3.10) 2.15.0 GPU (CUDA 12.1) 14 novembre 2024 14 novembre 2025 tf-2-15-cu121-VERSION_DATE-py310
2.14 (Python 3.10) 2.14.0 CPU uniquement 26 septembre 2024 26 septembre 2025 tf-2-14-cpu-VERSION_DATE-py310
2.14 (Python 3.10) 2.14.0 GPU (CUDA 11.8) 26 septembre 2024 26 septembre 2025 tf-2-14-cu118-VERSION_DATE-py310
2.13 (Python 3.10) 2.13.0 CPU uniquement 5 juillet 2024 5 juillet 2025 tf-2-13-cpu-VERSION_DATE-py310
2.13 (Python 3.10) 2.13.0 GPU (CUDA 11.8) 5 juillet 2024 5 juillet 2025 tf-2-13-cu118-VERSION_DATE-py310
2.12 (Python 3.10) 2.12.0 CPU uniquement 30 juin 2024 30 juin 2025 tf-2-12-cpu-VERSION_DATE-py310
2.12 (Python 3.10) 2.12.0 GPU (CUDA 11.8) 18 janvier 2024 18 janvier 2025 tf-2-12-cu113-VERSION_DATE-py310
2.11 (Python 3.10) 2.11.0 CPU uniquement 15 novembre 2023 15 novembre 2024 tf-2-11-cpu-VERSION_DATE-py310
2.11 (Python 3.10) 2.11.0 GPU (CUDA 11.3) 15 novembre 2023 15 novembre 2024 tf-2-11-cu113-VERSION_DATE-py310
2.11 2.11.0 CPU uniquement 15 novembre 2023 15 novembre 2024 tf-2-11-cpu-VERSION_DATE-py37
2.11 2.11.0 GPU (CUDA 11.3) 15 novembre 2023 15 novembre 2024 tf-2-11-cu113-VERSION_DATE-py37
2.10 2.10.1 CPU uniquement 15 novembre 2023 15 novembre 2024 tf-2-10-cpu-VERSION_DATE-py37
2.10 2.10.1 GPU (CUDA 11.3) 15 novembre 2023 15 novembre 2024 tf-2-10-cu113-VERSION_DATE-py37
2.9 2.9.3 CPU uniquement 15 novembre 2023 15 novembre 2024 tf-2-9-cpu-VERSION_DATE-py37
2.9 2.9.3 GPU (CUDA 11.3) 15 novembre 2023 15 novembre 2024 tf-2-9-cu113-VERSION_DATE-py37
2.8 2.8.4 CPU uniquement 15 novembre 2023 15 novembre 2024 tf-2-8-cpu-VERSION_DATE-py37
2.8 2.8.4 GPU (CUDA 11.3) 15 novembre 2023 15 novembre 2024 tf-2-8-cu113-VERSION_DATE-py37
2.6 (py39) 2.6.5 CPU uniquement 1er septembre 2023 1er septembre 2024 tf-2-6-cpu-VERSION_DATE-py39
2.6 (py39) 2.6.5 GPU (CUDA 11.3) 1er septembre 2023 1er septembre 2024 tf-2-6-cu110-VERSION_DATE-py39
2.6 (py37) 2.6.5 CPU uniquement 18 sept 2023 18 sept 2024 tf-2-6-cpu-VERSION_DATE-py37
2.6 (py37) 2.6.5 GPU (CUDA 11.3) 18 sept 2023 18 sept 2024 tf-2-6-cu110-VERSION_DATE-py37
2.3 2.3.4 CPU uniquement 1er septembre 2023 1er septembre 2024 tf-2-3-cpu
2.3 2.3.4 GPU (CUDA 11.3) 1er septembre 2023 1er septembre 2024 tf-2-3-cu110-VERSION_DATE

Versions de PyTorch

Version du framework de ML Version de correctif actuelle Accélérateurs compatibles Date de fin de correctif et de fin de compatibilité Date de fin de disponibilité Nom de la famille d'images
2.2 (Python 3.10) 2.2.0 CUDA 12.1 30 janvier 2025 30 janvier 2026 pytorch-2-2-VERSION_DATE-py310
2.1 (Python 3.10) 2.1.0 CUDA 12.1 4 octobre 2024 4 octobre 2025 pytorch-2-1-VERSION_DATE-py310
2.0 (Python 3.10) 2.0.0 CUDA 11.8 15 mars 2024 15 mars 2025 pytorch-2-0-VERSION_DATE-py310
1.13 (Python 3.10) 1.13.1 CUDA 11.3 8 décembre 2023 8 décembre 2024 pytorch-1-13-VERSION_DATE-py310
1.13 1.13.1 CUDA 11.3 8 décembre 2023 8 décembre 2024 pytorch-1-13-VERSION_DATE-py37
1.12 1.12.1 CUDA 11.3 1er septembre 2023 1er septembre 2024 pytorch-1-12-VERSION_DATE-py310

Étapes suivantes