Estruturas e regiões da AWS compatíveis - 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á.

Estruturas e regiões da AWS compatíveis

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

nota

A seleção inteligente do SageMaker oferece apoio ao treinamento de modelos do PyTorch com paralelismo de dados tradicional e paralelismo de dados distribuídos, o que cria réplicas de modelos em todos os funcionários 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 seleção inteligente do SageMaker funciona para trabalhos de paralelismo de dados, certifique-se de que o modelo que você treina caiba em cada memória da GPU.

Estruturas compatíveis

A seleção inteligente do SageMaker é compatível com os seguintes frameworks de aprendizado profundo e está disponível por meio dos contêineres de deep learning da AWS:

Tópicos

PyTorch

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

763.104.351.884.dkr.ecr.region.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker

Para ter mais informações sobre os contêineres predefinidos, consulte os contêineres do framework do SageMaker AI no AWS repositório do GitHub de contêineres de deep learning.

Regiões da AWS

Os contêineres empacotados com a biblioteca de seleção inteligente do SageMaker estão disponíveis nas Regiões da AWS onde os contêineres de deep learning da AWS estão em serviço.

Tipos de instância

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