Docker 注册表路径和示例代码 - Amazon SageMaker ECR 路径
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

Docker 注册表路径和示例代码

以下主题针对 Amazon SageMaker 提供的各个算法和深度学习容器 (DLC),列出其 Docker 注册表路径和其他参数。有关更多信息,请参阅使用预构建 SageMaker Docker 映像

如下所示使用路径:

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

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

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

注意

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

重要

检索 SageMaker XGBoost 映像 URI 时,请勿使用 :latest:1 作为映像 URI 标签。您必须指定一个支持的版本,这样才能选择由 SageMaker 管理且具有您要使用的原生 XGBoost 软件包版本的 XGBoost 容器。要查找迁移到 SageMaker XGBoost 容器中的软件包版本,请选择您的 Amazon Web Services 区域,然后导航到 XGBoost(算法)部分。

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