Amazon SageMaker AI 模型平行化程式庫 v2 範例 - Amazon SageMaker AI

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

Amazon SageMaker AI 模型平行化程式庫 v2 範例

此頁面提供部落格和 Jupyter 筆記本的清單,其中提供實作 SageMaker 模型平行化 (SMP) 程式庫 v2 以在 SageMaker AI 上執行分散式訓練任務的實際範例。

部落格與案例研究

下列部落格討論有關使用 SMP v2 的案例研究。

PyTorch 範例筆記本

範例筆記本在 SageMaker AI 範例 GitHub 儲存庫中提供。若要下載範例,請執行下列命令來複製儲存庫並前往 training/distributed_training/pytorch/model_parallel_v2

注意

在下列 SageMaker AI ML IDE 中複製並執行範例筆記本。

git clone https://github.com/aws/amazon-sagemaker-examples.git cd amazon-sagemaker-examples/training/distributed_training/pytorch/model_parallel_v2

SMP v2 範例筆記本