本文属于机器翻译版本。若本译文内容与英语原文存在差异，则一律以英文原文为准。
（已存档） SageMaker 模型并行度库 v1.x
重要
自 2023 年 12 月 19 日起， SageMaker 模型并行度 (SMP) 库 v2 已发布。为了支持 SMP 库 v2，未来版本将不再支持 SMP v1 功能。以下部分和主题已存档，专门用于使用 SMP 库 v1。有关使用 SMP 库 v2 的信息，请参阅 SageMaker 模型并行度库 v2。
使用 Amazon SageMaker AI 的模型并行库来训练由于 GPU 内存限制而难以训练的大型深度学习 (DL) 模型。该库可自动高效地将模型拆分为多个 GPUs 和实例。使用该库，您可以高效地训练具有数十亿或万亿参数的大型 DL 模型，从而更快地获得目标预测准确性。
您可以使用该库将自己的 PyTorch 模型 TensorFlow 和模型自动分成多个 GPUs 和多个节点，只需最少的代码更改。你可以通过 SageMaker Python 软件开发工具包访问该库的 API。
使用以下章节来了解有关模型并行性和模型并 SageMaker 行库的更多信息。该库的 API 文档位于 SageMaker Python SDK v2.199. 0 文档 APIs中的分布式训练