Imagens de estrutura e tipos de instância compatíveis Regiões da AWS - Amazon SageMaker
SageMaker imagens de estrutura pré-instaladas com SageMaker o ProfilerSageMaker Arquivos binários do pacote Profiler PythonSuportado 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 e tipos de instância compatíveis Regiões da AWS

Esse recurso é compatível com as seguintes estruturas de aprendizado de máquina e Regiões da AWS.

nota

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

SageMaker imagens de estrutura pré-instaladas com SageMaker o Profiler

SageMaker O Profiler está pré-instalado nos seguintes AWS Deep Learning Containers para. SageMaker

PyTorchimagens

PyTorch versões AWS DLCimagem URI
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

TensorFlow imagens

TensorFlow versões AWS DLCimagem URI
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 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 suportadas, se estiver usando versões anteriores da estrutura que não são mais suportadas.

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 Profiler SageMaker 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é-criados para PyTorch e TensorFlow, ou instale o pacote Profiler SageMaker Python localmente, use um dos seguintes arquivos binários. Dependendo do Python e das CUDA versões do seu ambiente, escolha uma das opções 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.

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

CPUe GPU criação de perfil

  • ml.g4dn.12xlarge

  • ml.g5.24xlarge

  • ml.g5.48xlarge

  • ml.p3dn.24xlarge

  • ml.p4de.24xlarge

  • ml.p4d.24xlarge

  • ml.p5.48xlarge

GPUsomente criação de perfil

  • ml.g5.2xlarge

  • ml.g5.4xlarge

  • ml.g5.8xlarge

  • ml.g5.16.xlarge