Cette page vous aide à choisir l'image de conteneur à utiliser.
Choisir un type d'image de conteneur
Les conteneurs de deep learning sont compatibles avec chaque version du framework et minimiser les failles de sécurité. Consulter le framework de conteneurs de deep learning de nos produits et services d'assistance pour comprendre les conséquences des dates de fin d'assistance et 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-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.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.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 conteneur Hugging Face
Cette section liste les visages câlins disponibles des images de conteneurs.
Conteneurs d'inférence pour la 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.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 de représentations vectorielles continues 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 |
---|---|---|---|---|---|
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.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 Transformer | 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 Transformer | 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 |
Familles d'images expérimentales
Les familles d'images qui ne sont pas répertoriées ici sont expérimentales. Image expérimentale les familles reçoivent une assistance au mieux, et ne reçoivent pas d'actualisations chaque nouvelle version du framework.
Répertorier toutes les versions disponibles
Si vous avez besoin d'un framework ou d'une version CUDA 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.