Scelta di un'immagine container

Questa pagina ti aiuta a scegliere l'immagine container 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 relative al supporto 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 contenitore appropriata, consulta le tabelle riportate di seguito.

Versioni di base

Versione framework ML Versione patch attuale Acceleratori supportati Data di fine del patch e del supporto Data di fine disponibilità Nome famiglia immagini
Base-Cu121 CUDA 12.1 (Python 3.10) CUDA 12.1 28 febbraio 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 gen 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° set 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 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° set 2023 1° settembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37

Versioni TensorFlow

Versione framework ML Versione patch attuale Acceleratori supportati Data di fine del patch e del supporto Data di fine disponibilità Nome famiglia immagini
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 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 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 luglio 2024 5 lug 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 nov 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 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 nov 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 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 novembre 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 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 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 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 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° 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° set 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° set 2023 1° set 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-3.py37

Versioni PyTorch

Versione framework ML Versione patch attuale 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 ottobre 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 mar 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° set 2023 1° settembre 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-12.py37

Immagini container Hugging Face

In questa sezione sono elencati i dispositivi Hugging Face disponibili. immagini container.

Contenitori di inferenza di generazione di testo

Versione framework Versione patch attuale 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

Contenitori di inferenza di incorporamenti di testo

Versione framework Versione patch attuale 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 patch attuale Acceleratori supportati Versione Transformers Data di fine della 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 patch attuale Acceleratori supportati Versione Transformers Data di fine della 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 patch attuale 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. Immagine sperimentale le famiglie sono supportate secondo il criterio del "best effort" e non ricevono aggiornamenti il a ogni nuova versione 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 opzioni disponibili per le immagini Deep Learning Containers, utilizza il seguente comando Google Cloud CLI con il tuo terminale preferito o Cloud Shell.

gcloud container images list --repository="us-docker.pkg.dev/deeplearning-platform-release/gcr.io"

Utilizzo locale

I container di deep learning possono essere estratti e utilizzati localmente. Per farlo, consulta la Guida introduttiva a un container di deep learning locale.

Passaggi successivi

  • Leggi la panoramica su Deep Learning Containers per scopri di più sugli elementi preinstallati nelle immagini 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.