Unterstützte Frameworks und AWS-Regionen - Amazon SageMaker KI

Die vorliegende Übersetzung wurde maschinell erstellt. Im Falle eines Konflikts oder eines Widerspruchs zwischen dieser übersetzten Fassung und der englischen Fassung (einschließlich infolge von Verzögerungen bei der Übersetzung) ist die englische Fassung maßgeblich.

Unterstützte Frameworks und AWS-Regionen

Bevor Sie SageMaker Smart Sifting beim Datenladeprozess verwenden, prüfen Sie, ob das Framework Ihrer Wahl unterstützt wird, ob die Instance-Typen in Ihrem AWS-Konto verfügbar sind und ob sich Ihr AWS-Konto in einem der unterstützten AWS-Regionen befindet.

Anmerkung

SageMaker Smart Sifting unterstützt das PyTorch-Modelltraining mit herkömmlicher Datenparallelität und verteilter Datenparallelität, wodurch Modellreplikate in allen GPU-Workern erstellt und die AllReduce-Operation verwendet wird. Es funktioniert nicht mit Methoden zur Modellparallelität wie der Parallelität fragmentierter Daten. Da SageMaker Smart Sifting für Datenparallelitätsaufträge funktioniert, sollten Sie sicherstellen, dass das Modell, das Sie trainieren, in jeden GPU-Speicher passt.

Unterstützte Frameworks

SageMaker Smart Sifting unterstützt die folgenden Deep-Learning-Frameworks und ist über Deep-Learning-Container von AWS erhältlich.

Themen

PyTorch

Framework Framework-Version Deep Learning Container URI
PyTorch 2.1.0

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

Weitere Informationen zu den vorgefertigten Containern finden Sie unter SageMaker AI Framework Containers im GitHub-Repository für Deep-Learning-Container von AWS.

AWS-Regionen

Die Container, die mit der Smart-Sifting-Bibliothek von SageMaker vorkonfiguriert sind, sind in den AWS-Regionen verfügbar, in denen Deep-Learning-Container von AWS ausgeführt werden.

Instance-Typen

Sie können SageMaker Smart Sifting für alle PyTorch-Trainingsjobs auf beliebigen Instance-Typen verwenden. Wir empfehlen die Verwendung von P4d-, P4de- oder P5-Instances.