调整自己的 Docker 容器以与 SageMaker 配合使用 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

调整自己的 Docker 容器以与 SageMaker 配合使用

您可以调整现有 Docker 映像以与 SageMaker 配合使用。如果您的容器满足某些功能或安全要求,而预构建的 SageMaker 映像当前不能满足这些要求,则您可能需要在 SageMaker 中使用现有的外部 Docker 映像。您可以通过两个工具来自带容器并对其进行调整,以便与 SageMaker 配合使用:

以下主题演示了如何使用 SageMaker 训练和推理工具包调整现有映像:

单个框架库

在 SageMaker 训练工具包和 SageMaker 推理工具包之外,SageMaker 还提供专门用于 TensorFlow、MXNet、PyTorch 和 Chainer 的工具包。下表提供的链接指向 GitHub 存储库,其中包含各个框架的源代码及其相应的服务工具包。链接的说明介绍了如何使用 Python SDK 在 SageMaker 上运行训练算法和托管模型。这些单独库的功能包含在 SageMaker 训练工具包和 SageMaker 推理工具包中。