本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
用于深度学习的预构建 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有关流程的概述,请参阅将您自己的预训练 MXnet TensorFlow 或模型带入亚马逊