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 al minimo le vulnerabilità di sicurezza. Esamina il framework Deep Learning Containers norme di assistenza per comprendere implicazioni delle date di fine assistenza e 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 patch attuale | Acceleratori supportati | Data di fine della patch e del supporto | Data di fine della disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|
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 lug 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 gen 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° 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 gen 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° set 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 della patch e del supporto | Data di fine della disponibilità | Nome famiglia immagini |
---|---|---|---|---|---|
2.15 (Python 3.10) | 2.15.0 | Solo CPU | 14 nov 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 nov 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 set 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 set 2024 | 26 set 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 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 nov 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 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 nov 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 nov 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° 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 della patch e del supporto | Data di fine della 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 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° set 2023 | 1° set 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-12.py37 |
Immagini container Hugging Face
In questa sezione sono elencate le funzionalità disponibili in Hugging Face. immagini container.
Contenitori di inferenza per la generazione di testo
Versione framework | Versione patch attuale | Acceleratori supportati | Data di fine della patch e del supporto | Data di fine della 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 patch attuale | Acceleratori supportati | Data di fine della patch e del supporto | Data di fine della 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 della 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 della 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 |
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 l'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 in locale
Deep Learning Containers può essere eseguito il pull e utilizzato localmente. A questo scopo, consulta Iniziare a utilizzare un deep learning locale container.
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.