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