本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。
带有 AI 的自定义 Docker 容 SageMaker 器
您可以调整现有的 Docker 镜像以与 SageMaker AI 配合使用。当您的容器满足预构建 SageMaker 的 SageMaker AI 镜像当前不支持的功能或安全要求时,您可能需要将现有的外部 Docker 镜像与 AI 配合使用。有两个工具包可以让你自带容器并对其进行调整以与 SageMaker AI 配合使用:
-
SageMaker 训练工具包
-使用此工具包使用 SageMaker AI 训练模型。 -
SageMaker AI 推理工具包
— 使用此工具包部署带有 SageMaker AI 的模型。
以下主题介绍如何使用 SageMaker 训练和推理工具包调整现有图像:
单个框架库
除了 SageMaker 训练工具包和 SageMaker 人工智能推理工具包外,A SageMaker I 还提供专门用于、 TensorFlow MXNet PyTorch、和 Chainer 的工具包。下表提供了指向 GitHub 存储库的链接,这些存储库包含每个框架的源代码及其各自的服务工具包。链接的说明用于使用 Python SDK 在 SageMaker AI 上运行训练算法和托管模型。这些独立库的功能包含在 SageMaker AI 训练工具包和 A SageMaker I 推理工具包中。
框架 | 工具包源代码 |
---|---|
TensorFlow |
|
MXNet |
|
PyTorch |
|
Chainer |