将 PyTorch 与 Amazon SageMaker 结合使用 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

将 PyTorch 与 Amazon SageMaker 结合使用

您可以通过 Amazon SageMaker,使用自定义 PyTorch 代码训练和部署模型。借助 SageMaker Python SDK PyTorch 估算器和模型以及 SageMaker 开源 PyTorch 容器,在 SageMaker 中编写和运行 PyTorch 脚本变得更加轻松。

您需要做什么?

我想要在 SageMaker 中训练自定义 PyTorch 模型。

有关示例 Jupyter 笔记本,请参阅 Amazon SageMaker 示例 GitHub 存储库中的 PyTorch 示例笔记本

有关文档,请参阅使用 PyTorch 训练模型

我有一个在 SageMaker 中训练过的 PyTorch 模型,我想将它部署到托管端点。

有关更多信息,请参阅部署 PyTorch 模型

我有一个在 SageMaker 之外训练过的 PyTorch 模型,我想将它部署到 SageMaker 端点

有关更多信息,请参阅部署您自己的 PyTorch 模型

我想查看 Amazon SageMaker Python SDK PyTorch 类的 API 文档。

有关更多信息,请参阅 PyTorch 类

我想查找 SageMaker PyTorch 容器存储库。

有关更多信息,请参阅 SageMaker PyTorch 容器 GitHub 存储库

我想查找有关 Amazon Deep Learning Containers 支持的 PyTorch 版本的信息。

有关更多信息,请参阅可用的深度学习容器映像

有关编写 PyTorch 训练脚本以及将 PyTorch 估算器和模型与 SageMaker 一起使用的一般信息,请参阅将 PyTorch 与 SageMaker Python SDK 结合使用