Amazon SageMaker AI 模型并行库版本 2 示例 - 亚马逊 SageMaker AI
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

Amazon SageMaker AI 模型并行库版本 2 示例

本页面提供了一个博客和 Jupyter Notebook 列表,其中介绍了实施 SageMaker 模型并行(SMP)库版本 2 以在 SageMaker AI 上运行分布式训练作业的实用示例。

博客和案例研究

以下博客将讨论有关使用 SMP 版本 2 的案例研究。

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 示例笔记本