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. 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 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

Versioni di TensorFlow

Versioni di PyTorch

Immagini container di Hugging Face

Questa sezione elenca i container di deep learning Hugging Face disponibili.

Contenitori di inferenza di generazione di testo

Contenitori di inferenza di incorporamenti di testo

Container di inferenza PyTorch

Container di addestramento PyTorch

Immagini container di Model Garden

Questa sezione elenca le immagini container di Model Garden disponibili.

Container di inferenza vLLM

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.