Usar o estimador genérico do SageMaker AI para estender contêineres predefinidos do DLC - SageMaker IA da Amazon

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

Usar o estimador genérico do SageMaker AI para estender contêineres predefinidos do DLC

Você pode personalizar esses contêineres predefinidos do SageMaker AI ou estendê-los para lidar com qualquer outro requisito funcional para seu algoritmo ou modelo ao qual a imagem predefinida do Docker do SageMaker AI não atenda. Para ver um exemplo de como você pode estender um contêiner pré-compilado, consulte Estender um contêiner pré-compilado.

Para estender um contêiner pré-compilado ou adaptar seu próprio contêiner para usar a biblioteca, você deve usar uma das imagens listadas em Frameworks compatíveis.

nota

Do TensorFlow 2.4.1 e do PyTorch 1.8.1, os DLCs do framework do SageMaker AI são compatíveis com tipos de instância habilitados para EFA. Recomendamos que você use as imagens DLC que contêm TensorFlow 2.4.1 ou posterior e PyTorch 1.8.1 ou posterior.

Por exemplo, se você usar PyTorch, seu Dockerfile deve conter uma declaração FROM semelhante à seguinte:

# SageMaker AI PyTorch image
FROM 763104351884.dkr.ecr.<aws-region>.amazonaws.com/pytorch-training:<image-tag>

ENV PATH="/opt/ml/code:${PATH}"

# this environment variable is used by the SageMaker AI PyTorch container to determine our user code directory.
ENV SAGEMAKER_SUBMIT_DIRECTORY /opt/ml/code

# /opt/ml and all subdirectories are utilized by SageMaker AI, use the /code subdirectory to store your user code.
COPY train.py /opt/ml/code/train.py

# Defines cifar10.py as script entrypoint
ENV SAGEMAKER_PROGRAM train.py

Você pode personalizar ainda mais seu próprio contêiner do Docker para trabalhar com o SageMaker AI usando o kit de ferramentas de treinamento do SageMaker AI e o arquivo binário da biblioteca de paralelismo de dados distribuídos do SageMaker AI. Para saber mais, consulte as instruções na seção a seguir.