Immagini del framework SageMaker AI preinstallate con SageMaker Profiler File binari del pacchetto Python di SageMaker Profiler Regioni AWS supportate Tipi di istanze supportati

Immagini del framework supportate, Regioni AWS e tipi di istanze

Questa funzionalità supporta i framework di machine learning e le Regioni AWS seguenti.

Nota

Per utilizzare questa funzionalità, assicurati di avere installato la versione 2.180.0 o successiva di SageMaker Python SDK.

Immagini del framework SageMaker AI preinstallate con SageMaker Profiler

SageMaker Profiler è disponibile nei seguenti container AWS di Deep Learning per SageMaker.

Immagini PyTorch

Versioni PyTorch	URI dell’immagine AWS DLC
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

Immagini TensorFlow

Versioni TensorFlow	URI dell’immagine AWS DLC
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

La distribuzione e la manutenzione dei container framework nelle tabelle precedenti sono soggette alla policy di supporto per i framework gestita dal servizio Container AWS per il Deep Learning. Se utilizzi versioni precedenti del framework che non sono più supportate, ti consigliamo fortemente di eseguire l’aggiornamento alle versioni del framework attualmente supportate.

Nota

Se desideri utilizzare SageMaker Profiler per altre immagini del framework o per tue immagini Docker, puoi installare SageMaker Profiler utilizzando i file binari del pacchetto Python di SageMaker Profiler forniti nella sezione seguente.

File binari del pacchetto Python di SageMaker Profiler

Se desideri configurare il tuo container Docker, utilizzare SageMaker Profiler in altri container predefiniti per PyTorch e TensorFlow o installare localmente il pacchetto Python di SageMaker Profiler, utilizza uno dei seguenti file binari. A seconda delle versioni di Python e CUDA nel tuo ambiente, scegli una delle seguenti opzioni.