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 |
|
Para ter mais informações sobre os contêineres predefinidos, consulte os contêineres do framework do SageMaker AI
Regiões da AWS
Os contêineres empacotados com a biblioteca de seleção inteligente do SageMaker
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.