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