Imagens de estruturas, Regiões da AWS e tipos de instâncias compatíveis - Amazon SageMaker
Imagens da estrutura do SageMaker pré-instaladas com o SageMaker ProfilerArquivos binários do pacote SageMaker Profiler PythonCompatível com Regiões da AWSTipos de instâncias compatíveis

Imagens de estruturas, Regiões da AWS e tipos de instâncias compatíveis

Esse atributo é compatível com as seguintes estruturas de machine learning e Regiões da AWS:

nota

Para usar esse atributo, certifique-se de ter instalada a versão 2.180.0 do SageMaker Python SDK ou versão posterior.

Imagens da estrutura do SageMaker pré-instaladas com o SageMaker Profiler

O SageMaker Profiler está pré-instalado nos seguintes contêineres de deep learning da AWS para o SageMaker:

Imagens do PyTorch

Versões PyTorch URI da imagem do DLC AWS
2.2.0

763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.2.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.1.0

763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.0.1

763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker

763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu121-ubuntu20.04-sagemaker
1.13.1

763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker

Imagens do TensorFlow

Versões do TensorFlow URI da imagem do DLC AWS
2.13.0

763104351884.dkr.ecr.<region>.amazonaws.com/tensorflow-training:2.13.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.12.0

763104351884.dkr.ecr.<region>.amazonaws.com/tensorflow-training:2.12.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.11.0

763104351884.dkr.ecr.<region>.amazonaws.com/tensorflow-training:2.11.0-gpu-py39-cu112-ubuntu20.04-sagemaker
Importante

A distribuição e a manutenção dos contêineres do framework nas tabelas anteriores estão sob a Política de Compatibilidade do Framework gerenciada pelo serviço de contêineres de deep learning da AWS. É altamente recomendável que você atualize para as versões da estrutura atualmente compatíveis, se estiver usando versões anteriores da estrutura que não são mais compatíveis.

nota

Se quiser usar o SageMaker Profiler para outras imagens de estrutura ou para suas próprias imagens do Docker, você pode instalar o SageMaker Profiler usando os arquivos binários do pacote Python do SageMaker Profiler fornecidos na seção a seguir.

Arquivos binários do pacote SageMaker Profiler Python

Se você quiser configurar seu próprio contêiner do Docker, use o SageMaker Profiler em outros contêineres pré-criados para PyTorch e TensorFlow, ou instale o pacote SageMaker Profiler Python localmente, use um dos seguintes arquivos binários: Dependendo das versões do Python e do CUDA em seu ambiente, escolha uma das opções apresentadas a seguir.

PyTorch

TensorFlow

Para obter mais informações sobre como instalar o SageMaker Profiler usando os arquivos binários, consulte (Opcional) Instalar o pacote SageMaker Profiler Python.

Compatível com Regiões da AWS

O SageMaker Profiler está disponível nas seguintes opções Regiões da AWS:

  • Leste dos EUA (Norte da Virgínia) (us-east-1)

  • Leste dos EUA (Ohio) (us-east-2)

  • Oeste dos EUA (Oregon) (us-west-2)

  • Europa (Frankfurt) (eu-central-1)

  • Europa (Irlanda) (eu-west-1)

Tipos de instâncias compatíveis

O SageMaker Profiler oferece apoio à criação de perfis de trabalhos de treinamento nos seguintes tipos de instância:

Perfil de CPU e GPU

  • ml.g4dn.12xlarge

  • ml.g5.24xlarge

  • ml.g5.48xlarge

  • ml.p3dn.24xlarge

  • ml.p4de.24xlarge

  • ml.p4d.24xlarge

  • ml.p5.48xlarge

Somente criação de perfil de GPU

  • ml.g5.2xlarge

  • ml.g5.4xlarge

  • ml.g5.8xlarge

  • ml.g5.16.xlarge