View a markdown version of this page

SageMaker Notes de mise à jour de la bibliothèque de parallélisme des données AI - Amazon SageMaker AI

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

SageMaker Notes de mise à jour de la bibliothèque de parallélisme des données AI

Consultez les notes de version suivantes pour suivre les dernières mises à jour de la bibliothèque SMDDP ( SageMaker AI Distributed Data Parallelism).

La bibliothèque de parallélisme des données distribuées SageMaker AI v2.5.0

Date : 17 octobre 2024

Nouvelles fonctionnalités

  • Ajout du support pour la PyTorch version 2.4.1 avec CUDA v12.1.

Intégration dans les conteneurs Docker distribués par la bibliothèque SageMaker AI model parallelism (SMP)

Cette version de la bibliothèque SMDDP est migrée vers La bibliothèque de parallélisme des SageMaker modèles v2.6.0.

658645717510.dkr.ecr.<us-west-2>.amazonaws.com/smdistributed-modelparallel:2.4.1-gpu-py311-cu121

Pour les régions dans lesquelles les images Docker SMP sont disponibles, consultez Régions AWS.

Fichier binaire de cette version

Vous pouvez télécharger ou installer la bibliothèque à l’aide de l’URL suivante.

https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.4.1/cu121/2024-10-09/smdistributed_dataparallel-2.5.0-cp311-cp311-linux_x86_64.whl

La bibliothèque de parallélisme des données distribuées SageMaker AI v2.3.0

Date : 11 juin 2024

Nouvelles fonctionnalités

  • Ajout du support pour la PyTorch version 2.3.0 avec CUDA v12.1 et Python v3.11.

  • Ajout du support pour PyTorch Lightning v2.2.5. Ceci est intégré dans le conteneur SageMaker AI Framework pour la PyTorch v2.3.0.

  • Ajout de la validation du type d’instance lors de l’importation pour empêcher le chargement de la bibliothèque SMDDP sur des types d’instance non pris en charge. Pour obtenir la liste des types d’instances compatibles avec la bibliothèque SMDDP, consultez Frameworks pris en charge, Régions AWS, et les types d'instances.

Intégration dans les conteneurs SageMaker AI Framework

Cette version de la bibliothèque SMDDP est migrée vers le conteneur SageMaker AI Framework suivant.

  • PyTorch v2.3.0

    763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.3.0-gpu-py311-cu121-ubuntu20.04-sagemaker

Pour obtenir la liste complète des versions de la bibliothèque SMDDP et des conteneurs prédéfinis, consultez Frameworks pris en charge, Régions AWS, et les types d'instances.

Fichier binaire de cette version

Vous pouvez télécharger ou installer la bibliothèque à l’aide de l’URL suivante.

https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.3.0/cu121/2024-05-23/smdistributed_dataparallel-2.3.0-cp311-cp311-linux_x86_64.whl

Autres modifications

  • La bibliothèque SMDDP v2.2.0 est intégrée au conteneur SageMaker AI Framework pour la v2.2.0. PyTorch

La bibliothèque de parallélisme des données distribuées SageMaker AI v2.2.0

Date : 04 mars 2024

Nouvelles fonctionnalités

  • Ajout du support pour la PyTorch v2.2.0 avec CUDA v12.1.

Intégration dans les conteneurs Docker distribués par la bibliothèque SageMaker AI model parallelism (SMP)

Cette version de la bibliothèque SMDDP est migrée vers La bibliothèque de parallélisme des SageMaker modèles v2.2.0.

658645717510.dkr.ecr.<region>.amazonaws.com/smdistributed-modelparallel:2.2.0-gpu-py310-cu121

Pour les régions dans lesquelles les images Docker SMP sont disponibles, consultez Régions AWS.

Fichier binaire de cette version

Vous pouvez télécharger ou installer la bibliothèque à l’aide de l’URL suivante.

https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.2.0/cu121/2024-03-04/smdistributed_dataparallel-2.2.0-cp310-cp310-linux_x86_64.whl

La bibliothèque de parallélisme des données distribuées SageMaker AI v2.1.0

Date : 1er mars 2024

Nouvelles fonctionnalités

  • Ajout du support pour la PyTorch v2.1.0 avec CUDA v12.1.

Corrections de bugs

  • Correction du problème de fuite de mémoire du CPU dans SMDDP v2.0.1.

Intégration dans les conteneurs SageMaker AI Framework

Cette version de la bibliothèque SMDDP a réussi les tests de référence et a été migrée vers le conteneur SageMaker AI Framework suivant.

  • PyTorch v2.1.0

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

Intégration dans les conteneurs Docker distribués par la bibliothèque SageMaker AI model parallelism (SMP)

Cette version de la bibliothèque SMDDP est migrée vers La bibliothèque de parallélisme des SageMaker modèles v2.1.0.

658645717510.dkr.ecr.<region>.amazonaws.com/smdistributed-modelparallel:2.1.2-gpu-py310-cu121

Pour les régions dans lesquelles les images Docker SMP sont disponibles, consultez Régions AWS.

Fichier binaire de cette version

Vous pouvez télécharger ou installer la bibliothèque à l’aide de l’URL suivante.

https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.1.0/cu121/2024-02-04/smdistributed_dataparallel-2.1.0-cp310-cp310-linux_x86_64.whl

La bibliothèque de parallélisme des données distribuées SageMaker AI v2.0.1

Date : 7 décembre 2023

Nouvelles fonctionnalités

Problèmes connus

  • Problème de fuite de mémoire du CPU dû à une augmentation progressive de la mémoire du CPU pendant l’entraînement avec l’opération AllReduce SMDDP en mode DDP.

Intégration dans les conteneurs SageMaker AI Framework

Cette version de la bibliothèque SMDDP a réussi les tests de référence et a été migrée vers le conteneur SageMaker AI Framework suivant.

  • PyTorch v2.0.1

    763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker

Fichier binaire de cette version

Vous pouvez télécharger ou installer la bibliothèque à l’aide de l’URL suivante.

https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.0.1/cu118/2023-12-07/smdistributed_dataparallel-2.0.2-cp310-cp310-linux_x86_64.whl

Autres modifications

  • À partir de cette version, la documentation de la bibliothèque SMDDP est entièrement disponible dans ce guide du développeur Amazon SageMaker AI. En faveur du guide complet du développeur pour SMDDP v2 contenu dans le Guide du développeur Amazon SageMaker AI, la documentation de référence supplémentaire pour SMDDP v1.x dans la documentation du SDK SageMaker AI Python n'est plus prise en charge. Si vous avez toujours besoin de la documentation SMP v1.x, consultez la capture d'écran suivante de la documentation sur SageMaker Python SDK v2.212.0 documentation.