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 al minimo le vulnerabilità di sicurezza. Leggi 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 container fornisce un ambiente Python 3 e include framework di data science selezionato (come PyTorch o TensorFlow), Conda, Stack NVIDIA per immagini GPU (CUDA, cuDNN, NCCL2) e molti altri pacchetti e strumenti di supporto. Per trovare l'immagine container appropriata, consulta le tabelle seguenti.

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 feb 2024 28 feb 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 lug 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° set 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 gen 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 lug 2025 11 lug 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 lug 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 nov 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 nov 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 set 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 lug 2024 5 lug 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 lug 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 nov 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 nov 2023 15 nov 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-11.py310
2,11 2.11.0 Solo CPU 15 nov 2023 15 nov 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-11.py37
2,11 2.11.0 CUDA 11.3 15 nov 2023 15 nov 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 nov 2023 15 nov 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 nov 2023 15 nov 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-10.py310
2,10 2.10.1 Solo CPU 15 nov 2023 15 nov 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-10.py37
2,10 2.10.1 CUDA 11.3 15 nov 2023 15 nov 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-10.py37
2,9 2.9.3 Solo CPU 15 nov 2023 15 nov 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-9.py37
2,9 2.9.3 CUDA 11.3 15 nov 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 nov 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 nov 2023 15 nov 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° set 2023 1° set 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° set 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-6.py37
2.3 2.3.4 Solo CPU 1° set 2023 1° set 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.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 mar 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 Hugging Face

Questa sezione elenca le immagini container 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.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

Incorporamenti di testo: container di inferenza

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

In questa sezione sono elencati i Model Garden disponibili. immagini container.

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 dei contenitori di deep learning 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 può essere eseguito il pull e utilizzato 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.
  • Inizia a utilizzare Deep Learning Containers esaminando le guide illustrative, che forniscono istruzioni su come creare ed eseguire il push di container di deep learning in formato Docker.