将 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 结合使用