Framework supportati e Regioni AWS - Amazon SageMaker AI

Le traduzioni sono generate tramite traduzione automatica. In caso di conflitto tra il contenuto di una traduzione e la versione originale in Inglese, quest'ultima prevarrà.

Framework supportati e Regioni AWS

Prima di utilizzare lo strumento di caricamento della funzionalità di smart sifting di SageMaker, verifica se il framework desiderato è supportato, se i tipi di istanza sono disponibili nell’account AWS e se l’account AWS figura tra le Regioni AWS supportate.

Nota

SageMaker Smart Sifting supporta l’addestramento dei modelli PyTorch con parallelizzazione dei dati tradizionale e distribuita, che crea repliche dei modelli in tutti i worker di GPU e utilizza l’operazione AllReduce. Non funziona con le tecniche di parallelizzazione dei modelli, inclusa la parallelizzazione dei dati con sharding. Poiché SageMaker Smart Sifting funziona per i processi di parallelizzazione dei dati, è opportuno verificare che il modello addestrato possa essere contenuto nella memoria di ciascuna GPU.

Framework supportati

SageMaker Smart Sifting supporta i seguenti framework di deep learning ed è disponibile tramite i Container AWS per il Deep Learning.

Argomenti

PyTorch

Framework Versione di Framework URI del container Deep Learning
PyTorch 2.1.0

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

Per ulteriori informazioni sui container predefiniti, consulta SageMaker AI Framework Containers nel repository GitHub di Container AWS per il Deep Learning.

Regioni AWS

I container forniti in pacchetto con la libreria SageMaker Smart Sifting sono disponibili nelle Regioni AWS in cui sono operativi i Container AWS per il Deep Learning.

Tipi di istanza

È possibile utilizzare SageMaker Smart Sifting per qualsiasi job di addestramento PyTorch su qualsiasi tipo di istanza. È consigliabile utilizzare istanze P4d, P4de o P5.