Questa pagina ti aiuta a scegliere l'immagine del contenitore da utilizzare.
Scegli un tipo di immagine container
Deep Learning Containers supporta ogni versione del framework in base a una pianificazione per minimizzare le vulnerabilità di sicurezza. Consulta le norme di assistenza del framework Deep Learning Containers per comprendere le implicazioni delle date di ritiro del supporto e di fine disponibilità.
Ogni immagine del contenitore fornisce un ambiente Python 3 e include il framework di scienza dei dati selezionato (ad esempio PyTorch o TensorFlow), Conda, lo stack NVIDIA per le immagini GPU (CUDA, cuDNN, NCCL2) e molti altri pacchetti e strumenti di supporto. Per trovare l'immagine contenitore appropriata, consulta le tabelle riportate di seguito.
Versioni di base
Versione framework ML | Versione attuale della patch | Acceleratori supportati | Data di fine del patch e del supporto | Data di fine disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|
Base-cu122 | CUDA 12.2 (Python 3.10) | CUDA 12.2 | 8 gennaio 2025 | 8 gennaio 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 febbraio 2024 | 28 febbraio 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu121.py310 |
Base-cu118 | CUDA 11.8 (Python 3.10) | CUDA 11.8 | 1° luglio 2024 | 1° luglio 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu118.py310 |
Base-cu113 (Python 3.10) | CUDA 11.3 | CUDA 11.3 | 1° gennaio 2024 | 1° gennaio 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu113.py310 |
Base-cu113 (Python 3.7) | CUDA 11.3 | CUDA 11.3 | 1° settembre 2023 | 1° settembre 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 | 1° gennaio 2024 | 1° gennaio 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py310 |
Base-cu110 (Python 3.7) | CUDA 11.0 | CUDA 11.0 | 1° settembre 2023 | 1° settembre 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37 |
Versioni di TensorFlow
Versione framework ML | Versione attuale della patch | Acceleratori supportati | Data di fine del patch e del supporto | Data di fine disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|
2.17 (Python 3.10) | 2.17.0 | Solo CPU | 11 luglio 2025 | 11 luglio 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 luglio 2025 | 11 luglio 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-17.py310 |
2.16 (Python 3.10) | 2.16.2 | Solo CPU | 28 giugno 2025 | 28 giugno 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 giugno 2025 | 28 giugno 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-16.py310 |
2.15 (Python 3.10) | 2.15.0 | Solo CPU | 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 | Solo CPU | 26 settembre 2024 | 26 settembre 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 settembre 2024 | 26 settembre 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-14.py310 |
2.13 (Python 3.10) | 2.13.0 | Solo CPU | 5 luglio 2024 | 5 luglio 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 luglio 2024 | 5 luglio 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-13.py310 |
2.12 (Python 3.10) | 2.12.0 | Solo CPU | 30 giugno 2024 | 30 giugno 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 giugno 2024 | 30 giugno 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-12.py310 |
2.11 (Python 3.10) | 2.11.0 | Solo CPU | 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 | Solo CPU | 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 | Solo CPU | 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 | Solo CPU | 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 | Solo CPU | 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 | Solo CPU | 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 | Solo CPU | 10 agosto 2023 | 10 agosto 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 agosto 2023 | 10 agosto 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-6.py39 |
2.6 (Python 3.7) | 2.6.5 | Solo CPU | 1° settembre 2023 | 1° settembre 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 | 1° settembre 2023 | 1° settembre 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-6.py37 |
2.3 | 2.3.4 | Solo CPU | 1° settembre 2023 | 1° settembre 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-3.py37 |
2.3 | 2.3.4 | CUDA 11.3 | 1° settembre 2023 | 1° settembre 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-3.py37 |
Versioni di PyTorch
Versione framework ML | Versione attuale della patch | Acceleratori supportati | Data di fine del patch e del supporto | Data di fine disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|
2.3 (Python 3.10) | 2.3.0 | CUDA 12.1 | 24 aprile 2025 | 24 aprile 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 gennaio 2025 | 30 gennaio 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 ott 2024 | 4 ottobre 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 marzo 2024 | 15 marzo 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 dicembre 2023 | 8 dicembre 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu118.1-13.py310 |
1,13 | 1.13.1 | CUDA 11.8 | 8 dicembre 2023 | 8 dicembre 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 dicembre 2023 | 8 dicembre 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu113.1-12.py310 |
1,12 | 1.12.1 | CUDA 11.3 | 1° settembre 2023 | 1° settembre 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-12.py37 |
Immagini container di Hugging Face
Questa sezione elenca i container di deep learning di Hugging Face disponibili.
Contenitori di inferenza di generazione di testo
Versione framework | Versione attuale della patch | Acceleratori supportati | Data di fine del patch e del supporto | Data di fine disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|
TGI 2.3 | 2.3.1 | CUDA 12.4 | 9 ottobre 2025 | 9 ottobre 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 luglio 2025 | 23 luglio 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 giugno 2025 | 27 giugno 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 aprile 2025 | 25 aprile 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 marzo 2025 | 13 marzo 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.1-4.ubuntu2204.py310 |
Contenitori di inferenza di incorporamenti di testo
Versione framework | Versione attuale della patch | Acceleratori supportati | Data di fine del patch e del supporto | Data di fine disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|
TEI 1.4 | 1.4.0 | CUDA 12.2 | 27 agosto 2025 | 27 agosto 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-4.ubuntu2204 |
TEI 1.4 | 1.4.0 | CPU | 27 agosto 2025 | 27 agosto 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 maggio 2025 | 2 maggio 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-2.ubuntu2204 |
TEI 1.2 | 1.2.2 | CPU | 2 maggio 2025 | 2 maggio 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-2 |
Container di inferenza PyTorch
Versione framework | Versione attuale della patch | Acceleratori supportati | Versione di Transformers | Data di fine del patch e del supporto | Data di fine disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|---|
PyTorch 2.2 | 2.2.2 | CUDA 12.1 | 4.41.1 | 12 giugno 2025 | 12 giugno 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 | CPU | 4.41.1 | 7 giugno 2025 | 7 giugno 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 agosto 2025 | 21 agosto 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 | CPU | 4.44.0 | 21 agosto 2025 | 21 agosto 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-44.ubuntu2204.py311 |
Container di addestramento PyTorch
Versione framework | Versione attuale della patch | Acceleratori supportati | Versione di Transformers | Data di fine del patch e del supporto | Data di fine disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|---|
PyTorch 2.3 | 2.3.0 | CUDA 12.1 | 4.42.3 | 21 agosto 2025 | 21 agosto 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-training-cu121.2-3.transformers.4-42.ubuntu2204.py310 |
Immagini container di Model Garden
Questa sezione elenca le immagini container di Model Garden disponibili.
Container di inferenza vLLM
Versione framework | Versione attuale della patch | Acceleratori supportati | Data di fine del patch e del supporto | Data di fine disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|
PyTorch 2.4 | 2.4.0 | CUDA 12.1 | 21 agosto 2025 | 21 agosto 2026 | us-docker.pkg.dev/deeplearning-platform-release/vertex-model-garden/vllm-inference.cu121.0-5.ubuntu2204.py310 |
Famiglie di immagini sperimentali
Le famiglie di immagini non elencate qui sono sperimentali. Le famiglie di immagini sperimentali sono supportate secondo il criterio del massimo impegno e non ricevono aggiornamenti con ogni nuova release del framework.
Elenco di tutte le versioni disponibili
Se hai bisogno di un framework o di una versione CUDA specifici, cerca nell'elenco completo delle immagini container disponibili. Per elencare tutte le immagini Deep Learning Containers disponibili, utilizza il seguente comando in Google Cloud CLI con il terminale che preferisci o in Cloud Shell.
gcloud container images list --repository="us-docker.pkg.dev/deeplearning-platform-release/gcr.io"
Utilizzo locale
Deep Learning Containers possono essere estratti e utilizzati localmente. Per farlo, consulta la Guida introduttiva a un container di deep learning locale.
Passaggi successivi
- Leggi la panoramica di Deep Learning Containers per scoprire di più su cosa è preinstallato nelle immagini dei container.
- Per iniziare a utilizzare Deep Learning Containers, consulta le guide pratiche, che forniscono istruzioni su come creare e inviare immagini dei container di deep learning.