Amazon SageMaker AI 模型平行處理程式庫 v1 範例 - Amazon SageMaker AI

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

Amazon SageMaker AI 模型平行處理程式庫 v1 範例

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

部落格與案例研究

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

範例筆記本

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

注意

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

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

PyTorch 的 SMP v1 範例筆記本

TensorFlow 的 SMP v1 範例筆記本