本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
将 PyTorch 与 Amazon SageMaker 结合使
您可以通过,使用自定义 PyTorch 代码训练和部署模型。借助 SageMaker Python 开发工具包 PyTorch 估算器和模型以及 SageMaker 开源 PyTorch 容器,在 SageMaker 中编写和运行 PyTorch 脚本变得更加轻松。
您需要做什么?
- 我想要在 SageMaker 中训练自定义 PyTorch 模型。
-
有关示例 Jupyter 笔记本,请参阅PyTorch 示例笔记本
在 Amazon SageMaker 示例 GitHub 存储库中。 有关文档,请参阅使用 PyTorch 训练模型
。 - 我有一个在 SageMaker 中训练过的 PyTorch 模型,我想将它部署到托管终端节点。
-
有关更多信息,请参阅 。部署 PyTorch 模型
. - 我有一个在 SageMaker 之外训练过的 PyTorch 模型,我想将它部署到 SageMaker 终端节点。
-
有关更多信息,请参阅 。从模型数据部署终端节点
. - 我想查看 API 文档Amazon SageMaker Python 开发工具包
PyTorch 类。 -
有关更多信息,请参阅 。PyTorch 类
. - 我想查找 SageMaker PyTorch 容器存储库。
-
有关更多信息,请参阅 。SageMaker PyTorch 容器 GitHub 库
. - 我想查找有关支持的 PyTorch 版本的信息AmazonDeep Learning Containers。
-
有关更多信息,请参阅 。可用的深度学习容器映像
.
有关编写 PyTorch 训练脚本以及将 PyTorch 估算器和模型与 SageMaker 结合使用的一般信息,请参将 PyTorch 与 SageMaker Python 开发工具包结合使用