Choisir une image de conteneur

Cette page vous aide à choisir l'image de conteneur à utiliser.

Choisir un type d'image de conteneur

Conteneurs de deep learning prend en charge chaque version de framework en fonction d'une programmation afin de minimiser les failles de sécurité. Consultez les règles de compatibilité du framework Deep Learning Containers pour comprendre les implications des dates de fin de période de compatibilité et de fin de disponibilité.

Chaque image de conteneur fournit un environnement Python 3 et inclut le framework de science de données sélectionné (tel que PyTorch ou TensorFlow), Conda, la pile NVIDIA pour les images GPU (CUDA, cuDNN, NCCL2), et bien d'autres formules d'assistance et outils. Pour trouver l'image de conteneur appropriée, consultez les tableaux ci-dessous.

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
Base-cu122 CUDA 12.2 (Python 3.10) CUDA 12.2 8 janvier 2025 8 janvier 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu122.py310
Base-cu121 CUDA 12.1 (Python 3.10) CUDA 12.1 28 février 2024 28 février 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu121.py310
Base-cu118 CUDA 11.8 (Python 3.10) CUDA 11.8 1er juillet 2024 1er juillet 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu118.py310
Base-cu113 (Python 3.10) CUDA 11.3 CUDA 11.3 1er janvier 2024 1er janvier 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu113.py310
Base-cu113 (Python 3.7) CUDA 11.3 CUDA 11.3 1er septembre 2023 1er septembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu113.py37
Base-cu110 (Python 3.10 / Debian 11) CUDA 11.0 CUDA 11.0 1er janvier 2024 1er janvier 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py310
Base-cu110 (Python 3.7) CUDA 11.0 CUDA 11.0 1er septembre 2023 1er septembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37

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.17 (Python 3.10) 2.17.0 CPU uniquement 11 juillet 2025 11 juillet 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-17.py310
2.17 (Python 3.10) 2.17.0 CUDA 12.3 11 juillet 2025 11 juillet 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-17.py310
2.16 (Python 3.10) 2.16.2 CPU uniquement 28 juin 2025 28 juin 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-16.py310
2.16 (Python 3.10) 2.16.2 CUDA 12.3 28 juin 2025 28 juin 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-16.py310
2.15 (Python 3.10) 2.15.0 CPU uniquement 14 novembre 2024 14 novembre 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-15.py310
2.15 (Python 3.10) 2.15.0 CUDA 12.1 14 novembre 2024 14 novembre 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu121.2-15.py310
2.14 (Python 3.10) 2.14.0 CPU uniquement 26 septembre 2024 26 septembre 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-14.py310
2.14 (Python 3.10) 2.14.0 CUDA 11.8 26 septembre 2024 26 septembre 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-14.py310
2.13 (Python 3.10) 2.13.0 CPU uniquement 5 juillet 2024 5 juillet 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-13.py310
2.13 (Python 3.10) 2.13.0 CUDA 11.8 5 juillet 2024 5 juillet 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-13.py310
2.12 (Python 3.10) 2.12.0 CPU uniquement 30 juin 2024 30 juin 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-12.py310
2.12 (Python 3.10) 2.12.0 CUDA 11.8 30 juin 2024 30 juin 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-12.py310
2.11 (Python 3.10) 2.11.0 CPU uniquement 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-11.py310
2.11 (Python 3.10) 2.11.0 CUDA 11.3 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-11.py310
2.11 2.11.0 CPU uniquement 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-11.py37
2.11 2.11.0 CUDA 11.3 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-11.py37
2.10 (Python 3.10) 2.10.1 CPU uniquement 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-10.py310
2.10 (Python 3.10) 2.10.1 CUDA 11.3 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-10.py310
2.10 2.10.1 CPU uniquement 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-10.py37
2.10 2.10.1 CUDA 11.3 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-10.py37
2.9 2.9.3 CPU uniquement 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-9.py37
2.9 2.9.3 CUDA 11.3 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-9.py37
2.8 2.8.4 CPU uniquement 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-8.py37
2.8 2.8.4 CUDA 11.3 15 novembre 2023 15 novembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-8.py37
2.6 (Python 3.9) 2.6.5 CPU uniquement 10 août 2023 10 août 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-6.py39
2.6 (Python 3.9) 2.6.5 CUDA 11.3 10 août 2023 10 août 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-6.py39
2.6 (Python 3.7) 2.6.5 CPU uniquement 1er septembre 2023 1er septembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-6.py37
2.6 (Python 3.7) 2.6.5 CUDA 11.3 1er septembre 2023 1er septembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-6.py37
2.3 2.3.4 CPU uniquement 1er septembre 2023 1er septembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-3.py37
2.3 2.3.4 CUDA 11.3 1er septembre 2023 1er septembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-3.py37

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.3 (Python 3.10) 2.3.0 CUDA 12.1 24 avril 2025 24 avril 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu121.2-3.py310
2.2 (Python 3.10) 2.2.0 CUDA 12.1 30 janvier 2025 30 janvier 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu121.2-2.py310
2.1 (Python 3.10) 2.1.0 CUDA 12.1 4 octobre 2024 4 octobre 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu121.2-1.py310
2.0 (Python 3.10) 2.0.0 CUDA 11.8 15 mars 2024 15 mars 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu118.2-0.py310
1.13 (Python 3.10) 1.13.1 CUDA 11.8 8 décembre 2023 8 décembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu118.1-13.py310
1.13 1.13.1 CUDA 11.8 8 décembre 2023 8 décembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-13.py37
1.12 (Python 3.10) 1.12.1 CUDA 11.3 8 décembre 2023 8 décembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu113.1-12.py310
1.12 1.12.1 CUDA 11.3 1er septembre 2023 1er septembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-12.py37

Images de conteneurs Hugging Face

Cette section répertorie les conteneurs de deep learning Hugging Face disponibles.

Conteneurs d'inférence de génération de texte

Version du framework 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
TGI 2.3 2.3.1 CUDA 12.4 9 octobre 2025 9 octobre 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu124.2-3.ubuntu2204.py311
TGI 2.2 2.2.0 CUDA 12.1 23 juillet 2025 23 juillet 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-2.ubuntu2204.py310
TGI 2.1 2.1.1 CUDA 12.1 27 juin 2025 27 juin 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-1.ubuntu2204.py310
TGI 2.0 2.0.1 CUDA 12.1 25 avril 2025 25 avril 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-0.ubuntu2204.py310
TGI 1.4 1.4.4 CUDA 12.1 13 mars 2025 13 mars 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.1-4.ubuntu2204.py310

Conteneurs d'inférence d'embedding textuel

Version du framework 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
TEI 1.4 1.4.0 CUDA 12.2 27 août 2025 27 août 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-4.ubuntu2204
TEI 1.4 1.4.0 Processeur 27 août 2025 27 août 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-4
TEI 1.2 1.2.2 CUDA 12.2 2 mai 2025 2 mai 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-2.ubuntu2204
TEI 1.2 1.2.2 Processeur 2 mai 2025 2 mai 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-2

Conteneurs d'inférence PyTorch

Version du framework Version de correctif actuelle Accélérateurs compatibles Version de Transformers Date de fin de correctif et de fin de compatibilité Date de fin de disponibilité Nom de la famille d'images
PyTorch 2.2 2.2.2 CUDA 12.1 4.41.1 12 juin 2025 12 juin 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-2.transformers.4-41.ubuntu2204.py311
PyTorch 2.2 2.2.2 Processeur 4.41.1 7 juin 2025 7 juin 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-41.ubuntu2204.py311
PyTorch 2.2 2.2.2 CUDA 12.1 4.44.0 21 août 2025 21 août 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-2.transformers.4-44.ubuntu2204.py311
PyTorch 2.2 2.2.2 Processeur 4.44.0 21 août 2025 21 août 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-44.ubuntu2204.py311

Conteneurs d'entraînement PyTorch

Version du framework Version de correctif actuelle Accélérateurs compatibles Version de Transformers Date de fin de correctif et de fin de compatibilité Date de fin de disponibilité Nom de la famille d'images
PyTorch 2.3 2.3.0 CUDA 12.1 4.42.3 21 août 2025 21 août 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-training-cu121.2-3.transformers.4-42.ubuntu2204.py310

Images de conteneurs Model Garden

Cette section répertorie les images de conteneurs Model Garden disponibles.

Conteneurs d'inférence vLLM

Version du framework 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
PyTorch 2.4 2.4.0 CUDA 12.1 21 août 2025 21 août 2026 us-docker.pkg.dev/deeplearning-platform-release/vertex-model-garden/vllm-inference.cu121.0-5.ubuntu2204.py310

Familles d'images expérimentales

Les familles d'images qui ne figurent pas dans cette liste sont expérimentales. Les familles d'images expérimentales sont prises en charge dans la mesure du possible et ne sont pas actualisées à chaque nouvelle version du framework.

Répertorier toutes les versions disponibles

Si vous avez besoin d'une version CUDA ou de framework spécifique, recherchez la liste complète des images de conteneurs disponibles. Pour répertorier toutes les images de conteneurs de deep learning disponibles, exécutez la commande suivante dans Google Cloud CLI avec un terminal de votre choix ou dans Cloud Shell.

gcloud container images list --repository="us-docker.pkg.dev/deeplearning-platform-release/gcr.io"

Utiliser des conteneurs en local

Les conteneurs de deep learning peuvent être extraits et utilisés localement. Pour ce faire, consultez la page Premiers pas avec un conteneur de deep learning local.

Étapes suivantes

  • Lisez la présentation des conteneurs de deep learning pour en savoir plus sur les éléments pré-installés sur les images de conteneur.
  • Pour commencer à utiliser les conteneurs de deep learning, consultez les guides d'utilisation qui fournissent des instructions sur la manière de créer et de transférer des images de conteneurs de deep learning.