Escolher uma imagem de contêiner

Veja nesta página dicas para escolher qual imagem de contêiner você quer usar.

Escolha um tipo de imagem de contêiner

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 do contêiner que você quer, consulte a tabela abaixo.

A lista a seguir de tipos de Deep Learning Containers é organizada por tipo de framework.

Framework Processador Nomes de imagens de contêiner
Base GPU gcr.io/deeplearning-platform-release/base-cu100
gcr.io/deeplearning-platform-release/base-cu101
gcr.io/deeplearning-platform-release/base-cu110
CPU gcr.io/deeplearning-platform-release/base-cpu
TensorFlow Enterprise 2.x GPU gcr.io/deeplearning-platform-release/tf2-gpu.2-1
gcr.io/deeplearning-platform-release/tf2-gpu.2-3
gcr.io/deeplearning-platform-release/tf2-gpu.2-6
CPU gcr.io/deeplearning-platform-release/tf2-cpu.2-1
gcr.io/deeplearning-platform-release/tf2-cpu.2-3
gcr.io/deeplearning-platform-release/tf2-cpu.2-6
TensorFlow Enterprise 1.x GPU gcr.io/deeplearning-platform-release/tf-gpu
gcr.io/deeplearning-platform-release/tf-gpu.1-15
CPU gcr.io/deeplearning-platform-release/tf-cpu
gcr.io/deeplearning-platform-release/tf-cpu.1-15
PyTorch GPU gcr.io/deeplearning-platform-release/pytorch-gpu
gcr.io/deeplearning-platform-release/pytorch-gpu.1-9
PyTorch XLA TPU/GPU/CPU (experimental) gcr.io/deeplearning-platform-release/pytorch-xla
gcr.io/deeplearning-platform-release/pytorch-xla.1-9
R CPU (experimental) gcr.io/deeplearning-platform-release/r-cpu
gcr.io/deeplearning-platform-release/r-cpu.4-0
Scikit-learn CPU (experimental) gcr.io/deeplearning-platform-release/sklearn-cpu
gcr.io/deeplearning-platform-release/sklearn-cpu.0-23
XGBoost CPU (experimenta) gcr.io/deeplearning-platform-release/xgboost-cpu
gcr.io/deeplearning-platform-release/xgboost-cpu.1-1
Imagens obsoletas do Deep Learning Container Misto gcr.io/deeplearning-platform-release/tf2-gpu.2-0
gcr.io/deeplearning-platform-release/tf2-cpu.2-0
gcr.io/deeplearning-platform-release/tf2-gpu.2-2
gcr.io/deeplearning-platform-release/tf2-cpu.2-2
gcr.io/deeplearning-platform-release/tf2-gpu.2-4
gcr.io/deeplearning-platform-release/tf2-cpu.2-4
gcr.io/deeplearning-platform-release/tf2-gpu.2-5
gcr.io/deeplearning-platform-release/tf2-cpu.2-5
gcr.io/deeplearning-platform-release/tf-gpu.1-13
gcr.io/deeplearning-platform-release/tf-gpu.1-14
gcr.io/deeplearning-platform-release/tf-cpu.1-13
gcr.io/deeplearning-platform-release/tf-cpu.1-14
gcr.io/deeplearning-platform-release/pytorch-gpu.1-0
gcr.io/deeplearning-platform-release/pytorch-gpu.1-1
gcr.io/deeplearning-platform-release/pytorch-gpu.1-2
gcr.io/deeplearning-platform-release/pytorch-gpu.1-3
gcr.io/deeplearning-platform-release/pytorch-gpu.1-4
gcr.io/deeplearning-platform-release/pytorch-gpu.1-6
gcr.io/deeplearning-platform-release/pytorch-gpu.1-7
gcr.io/deeplearning-platform-release/pytorch-gpu.1-8
gcr.io/deeplearning-platform-release/pytorch-cpu.1-0
gcr.io/deeplearning-platform-release/pytorch-cpu.1-1
gcr.io/deeplearning-platform-release/pytorch-cpu.1-2
gcr.io/deeplearning-platform-release/pytorch-cpu.1-3
gcr.io/deeplearning-platform-release/pytorch-cpu.1-4
gcr.io/deeplearning-platform-release/pytorch-xla.1-6
gcr.io/deeplearning-platform-release/pytorch-xla.1-7
gcr.io/deeplearning-platform-release/pytorch-xla.1-8

Imagens de contêiner do TensorFlow Enterprise

As imagens de contêiner do TensorFlow Enterprise oferecem a você uma distribuição do TensorFlow otimizada pelo Google Cloud. Versões específicas da distribuição do TensorFlow Enterprise também incluem suporte de versão de longo prazo. Para saber mais sobre o TensorFlow Enterprise, leia a visão geral do TensorFlow Enterprise.

Imagens experimentais

Algumas famílias de Deep Learning Containers são experimentais, conforme indicado pela tabela de famílias de imagens. Imagens experimentais são compatíveis com base no melhor esforço e podem não receber atualizações em cada nova versão do framework.

Como listar todas as versões disponíveis

Se você precisar de um framework ou uma versão CUDA específicos, pesquise a lista completa de imagens disponíveis. Para listar todas as imagens de Deep Learning Containers disponíveis, use o seguinte comando na ferramenta de linha de comando do gcloud com seu terminal preferido ou Cloud Shell.

gcloud container images list --repository="gcr.io/deeplearning-platform-release"

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.

A seguir