View a markdown version of this page

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

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á.

Imagens de estrutura suportadas, Regiões da AWS e tipos de instância

nota

Aviso de fim do suporte: em 30 de junho de 2027, o suporte para o Amazon SageMaker Profiler AWS será encerrado. Depois de 30 de junho de 2027, você não poderá mais acessar o console ou os recursos do Profiler. Para obter mais informações, consulte Alteração na disponibilidade do Profiler.

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

nota

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

SageMaker Imagens da estrutura de IA pré-instaladas com SageMaker o Profiler

SageMaker O Profiler está pré-instalado nos seguintes contêineres de aprendizado AWS profundo para SageMaker IA.

PyTorch imagens

PyTorch versões AWS URI da imagem DLC
2.2.0

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

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

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

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

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

TensorFlow imagens

TensorFlow versões AWS URI da imagem DLC
2.13.0

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

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

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

A distribuição e a manutenção dos contêineres da estrutura nas tabelas anteriores estão sob a Política de Suporte da Estrutura gerenciada pelo serviço AWS Deep Learning Containers. É 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 suas próprias imagens do Docker, você pode instalar o SageMaker Profiler usando os arquivos binários do pacote SageMaker Profiler Python fornecidos na seção a seguir.

SageMaker Arquivos binários do pacote Profiler Python

Se você quiser configurar seu próprio contêiner Docker, use o SageMaker Profiler em outros contêineres pré-construídos 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) Instale o pacote SageMaker Profiler Python.

Compatível Regiões da AWS

SageMaker O Profiler está disponível a seguir 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

SageMaker O Profiler oferece suporte à 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