Docker 注册表路径和示例代码 - 亚马逊 SageMaker ECR路径

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

Docker 注册表路径和示例代码

以下主题列出了亚马逊 SageMaker 提供的每种算法和 Deep Learning Containers (DLC) 的 Docker 注册路径和其他参数。有关更多信息,请参阅使用预构建的 SageMaker Docker 镜像

如下所示使用路径:

  • 要创建训练作业 (create_training_job),请指定训练映像的 Docker 注册表路径 (TrainingImage) 和训练输入模式 (TrainingInputMode)。您可以创建训练作业来通过特定数据集训练模型。

  • 要创建模型 (create_model),请为推理映像 () 指定 Docker 注册表路径 (Image)。PrimaryContainer Image SageMaker 启动基于端点配置的机器学习计算实例并部署模型,其中包括工件(模型训练的结果)。

  • 要创建模型监视器,请选择 AWS 区域,然后选择模型监视器(算法)。有关更多信息,请参阅 Amazon SageMaker 模型监控器预建容器

注意

对于注册表路径,请使用 :1 版本标签以确保您使用了算法/DLC 的稳定版本。您可以通过在具有 :1 标签的推理镜像上使用具有 :1 标签的镜像来可靠地托管已训练的模型。在注册表路径中使用:latest标签可以获得算法/DLC 的最高 up-to-date 版本,但可能会导致向后兼容性问题。避免将 :latest 标签用于生产用途。

重要

检索 SageMaker XGBoost 图像 URI 时,请勿使用:latest:1作为图像 URI 标签。您必须指定一个支持的版本才能选择包含要使用的本机 XGBoost 包版本的 SageMaker托管 XGBoost 容器。要查找迁移到 SageMaker XGBoost 容器中的软件包版本,请选择您的, AWS 区域 然后导航到 XGBoost(算法)部分。

要查找注册表路径,请选择 AWS 区域,然后选择算法或 DLC。