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.