使用 SageMaker AI 分散式資料平行化程式庫進行分散式訓練 - Amazon SageMaker AI

本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。

使用 SageMaker AI 分散式資料平行化程式庫進行分散式訓練

SageMaker AI 分散式資料平行化 (SMDDP) 程式庫旨在簡化使用,並提供與 PyTorch 的無縫整合。

在 SageMaker AI 使用 SMDDP 程式庫訓練深度學習模型時,您可以專注於撰寫訓練指令碼和模型訓練。

若要開始使用,請匯入 SMDDP 程式庫,以使用其針對 AWS 最佳化的集體操作。下列主題會根據您要最佳化的集體操作,提供如何新增至訓練指令碼的指示。