Especificações de SageMaker imagem personalizadas

Importante

Em 30 de novembro de 2023, a experiência anterior do Amazon SageMaker Studio agora se chama Amazon SageMaker Studio Classic. A seção a seguir é específica para usar o aplicativo Studio Classic. Para obter informações sobre como usar a experiência atualizada do Studio, consulte SageMaker Estúdio Amazon.

As especificações a seguir se aplicam à imagem do contêiner representada por uma versão SageMaker da imagem.

Executando a imagem

ENTRYPOINTe CMD as instruções são substituídas para permitir que a imagem seja executada como um KernelGateway aplicativo.

A porta 8888 na imagem está reservada para executar o servidor KernelGateway web.

Interrompendo a imagem

Os DeleteApp API problemas são equivalentes a um docker stop comando. Outros processos no contêiner não receberão os SIGTERM sinais SIGKILL /.

Descoberta do kernel

SageMaker reconhece os kernels conforme definido pelas especificações do kernel do Jupyter.

Você pode especificar uma lista de kernels a serem exibidos antes de executar a imagem. Se não for especificado, python3 será exibido. Use o DescribeAppImageConfigAPIpara ver a lista de kernels.

Os ambientes Conda são reconhecidos como especificações do kernel por padrão.

Sistema de arquivos

Os diretórios /opt/.sagemakerinternal e /opt/ml são reservados. Qualquer dado nesses diretórios pode não estar visível em runtime.

Dados do usuário

Cada usuário em um domínio obtém um diretório de usuários em um volume compartilhado do Amazon Elastic File System na imagem. A localização do diretório do usuário atual no EFS volume da Amazon é configurável. Por padrão, o local do diretório é /home/sagemaker-user.

SageMaker configura POSIXUID/GIDmapeamentos entre a imagem e o host. O padrão é mapear oUID/GID(0/0) do usuário root para oUID/GIDno host.

Você pode especificar esses valores usando CreateAppImageConfigAPIo.

GID/UIDlimites

O Amazon SageMaker Studio Classic só oferece suporte ao seguinte DefaultUID e às DefaultGID combinações a seguir:

• PadrãoUID: 1000 e PadrãoGID: 100, o que corresponde a um usuário sem privilégios.

• PadrãoUID: 0 e PadrãoGID: 0, que corresponde ao acesso root.

Metadados

Um arquivo de metadados está localizado em /opt/ml/metadata/resource-metadata.json. Nenhuma variável de ambiente adicional é incluída às variáveis definidas na imagem. Para obter mais informações, consulte Obter metadados do aplicativo.

GPU

Em uma GPU instância, a imagem é executada com a --gpus opção. Somente o CUDA kit de ferramentas deve ser incluído na imagem, não os NVIDIA drivers. Para obter mais informações, consulte o Guia NVIDIA do usuário.

Métricas e registro em log

Os registros do KernelGateway processo são enviados para a Amazon CloudWatch na conta do cliente. O nome do grupo de logs é /aws/sagemaker/studio. O nome do fluxo de logs é $domainID/$userProfileName/KernelGateway/$appName.

Tamanho da imagem

Limitado a 35 GB. Para ver o tamanho da sua imagem, execute docker image ls.

Exemplo de Dockerfile

O exemplo de Dockerfile a seguir cria uma imagem baseada no Amazon Linux 2, instala pacotes de terceiros e o python3 kernel e define o escopo para o usuário não privilegiado.

FROM public.ecr.aws/amazonlinux/amazonlinux:2

ARG NB_USER="sagemaker-user"
ARG NB_UID="1000"
ARG NB_GID="100"

RUN \
    yum install --assumeyes python3 shadow-utils && \
    useradd --create-home --shell /bin/bash --gid "${NB_GID}" --uid ${NB_UID} ${NB_USER} && \
    yum clean all && \
    python3 -m pip install ipykernel && \
    python3 -m ipykernel install

USER ${NB_UID}