View a markdown version of this page

Frameworks compatíveis e AWS Regiões - 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á.

Frameworks compatíveis e AWS Regiões

Antes de usar o carregador de dados de filtragem SageMaker inteligente, verifique se sua estrutura de escolha é compatível, se os tipos de instância estão disponíveis em sua AWS conta e se sua AWS conta está em uma das regiões compatíveis. AWS

nota

SageMaker a peneiração inteligente suporta o treinamento de PyTorch modelos com paralelismo de dados tradicional e paralelismo de dados distribuído, o que cria réplicas de modelos em todos os trabalhadores da GPU e usa a operação. AllReduce Ele não funciona com técnicas de paralelismo de modelos, incluindo paralelismo de dados fragmentados. Como a filtragem SageMaker inteligente funciona para trabalhos de paralelismo de dados, certifique-se de que o modelo que você treina caiba na memória de cada GPU.

Estruturas compatíveis

SageMaker O smart sifting suporta as seguintes estruturas de aprendizado profundo e está disponível por meio de contêineres de aprendizado AWS profundo.

Tópicos

PyTorch

Framework Versão do framework Contêineres de deep learning
PyTorch 2.1.0

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

Para obter mais informações sobre os contêineres pré-criados, consulte Contêineres do SageMaker AI Framework no GitHub repositório de contêineres de aprendizado AWS profundo.

Regiões da AWS

Os contêineres embalados com a biblioteca de peneiração SageMaker inteligente estão disponíveis Regiões da AWS onde os contêineres de aprendizado AWS profundo estão em serviço.

Tipos de instância

Você pode usar a filtragem SageMaker inteligente para qualquer trabalho de PyTorch treinamento em qualquer tipo de instância. Recomendamos que você use instâncias P4d, P4de ou P5.