Scelta di un'immagine container

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.