本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
用于深度学习的预构建 SageMaker Docker 镜像
Amazon SageMaker 提供预构建的 Docker 镜像,其中包括深度学习框架以及训练和推理所需的其他依赖项。有关由管理的预构建 Docker 镜像的完整列表 SageMaker,请参阅 Docker 注册表路径和示例代码。
使用 SageMaker Python 开发工具包
借助 SageMaker Python SDK
框架 | 说明 |
---|---|
TensorFlow |
|
MXNet |
|
PyTorch |
|
Chainer |
|
Hugging Face |
扩展预构建的 SageMaker Docker 镜像
您可以自定义这些预构建的容器或对其进行扩展,以满足预构建 SageMaker Docker 镜像不支持的算法或模型的任何其他功能要求。有关示例,请参阅通过扩展现有 PyTorch 容器, SageMaker 使用您自己的脚本和数据集微调和部署 BerTopic 模型
您还可以使用预构建的容器来部署您的自定义模型或已在其他框架中训练过的模型。 SageMaker要概述将经过训练的模型工件导入终端节点 SageMaker 并将其托管到终端节点的过程,请参阅将您自己的预训练的 MXnet TensorFlow 或模型带入