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

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

Docker 注册表路径和示例代码

以下主题列出了每个 Amazon SageMaker 提供的算法和 Deep Learning Containers (DLC) 的 Docker 注册表路径和其他参数。

使用路径,如下所示:

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

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

注意

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

对于 XGBoost,请不要使用:latest或者:1. 使用您需要的特定版本,例如:0.90-1-cpu-py3:0.90-2-cpu-py3:1.0-1-cpu-py3,或者:1.2-1.

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