Criar um contêiner derivado

Nesta página, descrevemos como criar um contêiner derivado com base em uma das imagens padrão de Deep Learning Containers disponíveis.

Para concluir as etapas deste guia, é possível usar o Cloud Shell ou qualquer ambiente com o Google Cloud CLI instalado.

Antes de começar

Antes de começar, conclua as etapas a seguir.

  1. Conclua as etapas de configuração na seção "Primeiros passos" em Primeiros passos com um contêiner de aprendizado profundo local.

  2. Verifique se a cobrança está ativada para o seu projeto do Google Cloud.

    Saiba como ativar o faturamento

  3. Ative a API Artifact Registry.

    Ativar a API

O processo

Para criar um contêiner derivado, você usará um processo semelhante a este:

  1. Crie o Dockerfile inicial e execute comandos de modificação.

    Para começar, crie um contêiner Deep Learning Containers usando um dos tipos de imagem disponíveis. Em seguida, use os comandos conda, pip ou Jupyter para modificar a imagem do contêiner de acordo com suas necessidades.

  2. Crie e envie a imagem do contêiner.

    Crie a imagem do contêiner e, em seguida, envie-a para um lugar acessível para sua conta de serviço do Compute Engine.

Criar o Dockerfile inicial e executar comandos de modificação

Use os seguintes comandos para selecionar um tipo de imagem do Deep Learning Containers e fazer uma pequena alteração na imagem do contêiner. Neste exemplo, mostramos como começar com uma imagem do TensorFlow e atualizá-la com a versão mais recente do TensorFlow. Escreva os seguintes comandos no Dockerfile:

FROM us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu:latest
# Uninstall the container's TensorFlow version and install the latest version
RUN pip uninstall -y tensorflow && \
    pip install tensorflow

Crie e envie a imagem do contêiner

Use os comandos a seguir para criar e enviar a imagem do contêiner para o Artifact Registry, onde ela pode ser acessada pela conta de serviço do Google Compute Engine.

Crie e autentique o repositório:

export PROJECT=$(gcloud config list project --format "value(core.project)")
gcloud artifacts repositories create REPOSITORY_NAME \
    --repository-format=docker \
    --location=LOCATION
gcloud auth configure-docker LOCATION-docker.pkg.dev

Substitua:

  • LOCATION: o local regional ou multirregional do repositório, por exemplo, us. Para ver uma lista de locais compatíveis, execute o comando gcloud artifacts locations list.
  • REPOSITORY_NAME: o nome do repositório que você quer criar, por exemplo, my-tf-repo.

Em seguida, crie e envie a imagem:

export IMAGE_NAME="LOCATION-docker.pkg.dev/${PROJECT}/REPOSITORY_NAME/tf-custom:v1"
docker build . -t $IMAGE_NAME
docker push $IMAGE_NAME