Escolher uma imagem de contêiner

Esta página ajuda você a escolher qual imagem de contêiner usar.

Escolha um tipo de imagem de contêiner

O Deep Learning Containers é compatível com cada versão de framework com base em uma programação para minimizar as vulnerabilidades de segurança. Leia a política de suporte ao framework do Deep Learning Containers para entender as implicações das datas de término de suporte e fim da disponibilidade.

Cada imagem de contêiner oferece um ambiente Python 3 e inclui o framework de ciência de dados selecionado (como PyTorch ou TensorFlow), Conda, a pilha NVIDIA para imagens de GPU (CUDA, cuDNN, NCCL2) e muitas outras ferramentas e pacotes de suporte. Para encontrar a imagem de contêiner apropriada, consulte a tabela abaixo.

Versões de base

Versões do TensorFlow

Versões do PyTorch

Imagens de contêiner do Model Garden

Esta seção lista as imagens de contêiner disponíveis do Model Garden.

Contêineres de inferência vLLM

Famílias de imagens experimentais

As famílias de imagens que não estão listadas aqui são experimentais. As famílias de imagens experimentais são compatíveis com base no melhor esforço e não recebem atualizações em cada nova versão do framework.

Como listar todas as versões disponíveis

Se você precisar de um framework ou versão CUDA específicos, pesquise a lista completa de imagens de contêiner disponíveis. Para listar todas as imagens de contêineres de aprendizado profundo disponíveis, use o seguinte comando na Google Cloud CLI com seu terminal preferido ou no Cloud Shell.

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

Como usar localmente

Os Deep Learning Containers podem ser extraídos e usados localmente. Para fazer isso, consulte Primeiros passos com um Deep Learning Container local.

Ver ocorrências de vulnerabilidade

Para ver as ocorrências de vulnerabilidade de uma imagem dos Deep Learning Containers, consulte Ver ocorrências usando gcloud.

A seguir

• Leia a Visão geral dos Deep Learning Containers para saber mais sobre o que está pré-instalado nas imagens de contêiner.
• Para começar a usar Deep Learning Containers, leia os guias de instruções, que oferecem instruções sobre como criar e enviar imagens do Deep Learning Containers.