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