本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
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 中克隆并运行示例笔记本。
-
SageMaker JupyterLab(在 2023 年 12 月之后创建的 Studio 中可用)
-
SageMaker Code Editor(在 2023 年 12 月之后创建的 Studio 中可用)
-
Studio Classic(可作为 2023 年 12 月之后创建的 Studio 中的应用程序使用)
git clone https://github.com/aws/amazon-sagemaker-examples.git cd amazon-sagemaker-examples/training/distributed_training/pytorch/model_parallel_v2
SMP v2 示例笔记本