Immagini di framework e tipi Regioni AWS di istanze supportati - Amazon SageMaker
SageMaker immagini del framework preinstallate con Profiler SageMaker SageMaker File binari del pacchetto Python di ProfilerSupportato Regioni AWSTipi di istanze supportati

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à.

Immagini di framework e tipi Regioni AWS di istanze supportati

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

Nota

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

SageMaker immagini del framework preinstallate con Profiler SageMaker

SageMaker Profiler è preinstallato nei seguenti AWS Deep Learning Containers per. SageMaker

PyTorchimmagini

PyTorch versioni AWS DLCimmagine URI
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

TensorFlow immagini

TensorFlow versioni AWS DLCimmagine URI
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 contenitori del framework nelle tabelle precedenti sono soggette alla Framework Support Policy gestita dal servizio AWS Deep Learning Containers. Ti consigliamo vivamente di eseguire l'aggiornamento alle versioni del framework attualmente supportate, se utilizzi versioni precedenti del framework che non sono più supportate.

Nota

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

SageMaker File binari del pacchetto Python di Profiler

Se desideri configurare il tuo contenitore Docker, utilizzare SageMaker Profiler in altri contenitori predefiniti per PyTorch e TensorFlow, oppure installare il pacchetto Profiler SageMaker Python localmente, usa uno dei seguenti file binari. A seconda del Python e delle CUDA versioni del tuo ambiente, scegli una delle seguenti opzioni.

PyTorch

TensorFlow

Per ulteriori informazioni su come installare SageMaker Profiler utilizzando i file binari, vedere. (Facoltativo) Installa il pacchetto SageMaker Profiler Python

Supportato Regioni AWS

SageMaker Profiler è disponibile nelle seguenti Regioni AWS versioni.

  • Stati Uniti orientali (Virginia settentrionale) (us-east-1)

  • Stati Uniti orientali (Ohio) (us-east-2)

  • Stati Uniti occidentali (Oregon) (us-west-2)

  • Europa (Francoforte) (eu-central-1)

  • Europa (Irlanda) (eu-west-1)

Tipi di istanze supportati

SageMaker Profiler supporta la profilazione dei lavori di formazione sui seguenti tipi di istanze.

CPUe profilazione GPU

  • ml.g4dn.12xlarge

  • ml.g5.24xlarge

  • ml.g5.48xlarge

  • ml.p3dn.24xlarge

  • ml.p4de.24xlarge

  • ml.p4d.24xlarge

  • ml.p5.48xlarge

GPUsolo profilazione

  • ml.g5.2xlarge

  • ml.g5.4xlarge

  • ml.g5.8xlarge

  • ml.g5.16.xlarge