AWS SageMaker - 上传我们自己的 docker 图像

Question

我是 AWS SageMaker 的新手，我正在使用这项技术来构建和训练机器学习模型。 我现在开发了一个 docker 图像，其中包含我们为 tensorflow 的自定义代码。 我想将此自定义 docker 图像上传到 AWS SageMaker 并使用它。

我搜索了各种链接，但找不到有关如何上传我们自己的自定义 docker 图像的正确信息。

您能否建议我有关将我们自己的 docker 图像上传到 AWS SageMaker 的过程的推荐链接？

Answer 1

为了使用 sagemaker，您必须将容器推送到ECR 。 最重要的是容器必须“适应”以适应 sagemaker 的要求，但这里描述了一切。 此外，如果您想看一个示例，这里是我的.. 我在 AWS Sagemaker 中将我的容器与 TF Object 检测 API 一起使用。