调整自己的 Docker 容器以与 SageMaker 配合使用
您可以调整现有 Docker 映像以与 SageMaker 配合使用。如果您的容器满足某些功能或安全要求,而预构建的 SageMaker 映像当前不能满足这些要求,则您可能需要在 SageMaker 中使用现有的外部 Docker 映像。您可以通过两个工具来自带容器并对其进行调整,以便与 SageMaker 配合使用:
以下主题演示了如何使用 SageMaker 训练和推理工具包调整现有映像:
单个框架库
在 SageMaker 训练工具包和 SageMaker 推理工具包之外,SageMaker 还提供专门用于 TensorFlow、MXNet、PyTorch 和 Chainer 的工具包。下表提供的链接指向 GitHub 存储库,其中包含各个框架的源代码及其相应的服务工具包。链接的说明介绍了如何使用 Python SDK 在 SageMaker 上运行训练算法和托管模型。这些单独库的功能包含在 SageMaker 训练工具包和 SageMaker 推理工具包中。
框架 | 工具包源代码 |
---|---|
TensorFlow |
|
MXNet |
|
PyTorch |
|
Chainer |