Docker 注册表路径和 (ap-east-2) 的示例代码 - ECR 路径

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

Docker 注册表路径和 (ap-east-2) 的示例代码

以下主题列出了 Amazon A SageMaker I 在本文中提供的每种算法和深度学习容器的参数 AWS 区域。

PyTorch (DLC)

有关支持和不支持的 PyTorch 版本的信息,请参阅 Dee AWS p Learning Containers 开发者指南中的框架支持策略表

以下 SageMaker AI Python SDK 示例展示了如何检索特定的注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='pytorch',region='ap-east-2',version='1.8.0',py_version='py3',image_scope='inference', instance_type='ml.c5.4xlarge')
注册表路径 版本 作业类型(映像范围) 处理器类型 Python 版本
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 2.7.1 训练 CPU,GPU py312
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 2.6.0 推理 CPU,GPU py312
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 2.6.0 训练 CPU,GPU py312
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 2.5.1 推理 CPU,GPU py311
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 2.5.1 训练 CPU,GPU py311
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 2.4.0 推理 CPU,GPU py311
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> 2.4.0 inference_graviton CPU py311
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 2.4.0 训练 CPU,GPU py311
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 2.3.0 推理 CPU,GPU py311
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> 2.3.0 inference_graviton CPU py311
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 2.3.0 训练 CPU,GPU py311
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> 2.2.1 inference_graviton CPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 2.2.0 推理 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 2.2.0 训练 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 2.1.0 推理 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> 2.1.0 inference_graviton CPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 2.1.0 训练 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 2.0.1 推理 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> 2.0.1 inference_graviton CPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 2.0.1 训练 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 2.0.0 推理 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> 2.0.0 inference_graviton CPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 2.0.0 训练 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.13.1 推理 CPU,GPU py39
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.13.1 训练 CPU,GPU py39
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.12.1 推理 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> 1.12.1 inference_graviton CPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.12.1 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.12.0 推理 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.12.0 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.11.0 推理 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.11.0 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.10.2 推理 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.10.2 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.10.0 推理 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.10.0 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.9.1 推理 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.9.1 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.9.0 推理 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.9.0 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.8.1 推理 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.8.1 训练 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.8.0 推理 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.8.0 训练 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.7.1 推理 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.7.1 训练 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.6.0 推理 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.6.0 训练 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.5.0 推理 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.5.0 训练 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.4.0 推理 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.4.0 训练 CPU,GPU py2、py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.3.1 推理 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.3.1 训练 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> 1.2.0 推理 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> 1.2.0 训练 CPU,GPU py2、py3

Spark(算法)

以下 SageMaker AI Python SDK 示例展示了如何检索特定的注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='spark',region='ap-east-2',version='3.0',image_scope='processing')
注册表路径 版本 作业类型(映像范围)
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> 3.3 处理
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> 3.2 处理
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> 3.1 处理
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> 3.0 处理
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> 2.4 处理

Tensorflow (DLC)

有关支持和不支持的 TensorFlow 版本的信息,请参阅 Dee AWS p Learning Containers 开发者指南中的框架支持策略表

以下 SageMaker AI Python SDK 示例展示了如何检索特定的注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='tensorflow',region='ap-east-2',version='1.12.0',image_scope='inference',instance_type='ml.c5.4xlarge')
注册表路径 版本 作业类型(映像范围) 处理器类型 Python 版本
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.19.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.19.0 训练 CPU,GPU py312
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.18.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.18.0 训练 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.16.2 训练 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.16.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> 2.16.1 inference_graviton CPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.14.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> 2.14.1 inference_graviton CPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.14.1 训练 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.13.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> 2.13.0 inference_graviton CPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.13.0 训练 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.12.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> 2.12.1 inference_graviton CPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.12.0 训练 CPU,GPU py310
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.11.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.11.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.11.0 训练 CPU,GPU py39
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.10.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.10.1 训练 CPU,GPU py39
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.10.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.9.3 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.9.2 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.9.2 训练 CPU,GPU py39
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> 2.9.1 inference_graviton CPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.8.4 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.8.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.8.0 训练 CPU,GPU py39
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.7.1 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.7.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.6.3 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.6.3 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.6.2 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.6.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.6.0 训练 CPU,GPU py38
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.5.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.5.1 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.5.0 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.4.3 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.4.3 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.4.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.4.1 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.3.2 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.3.2 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.3.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.3.1 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.3.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.3.0 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.2.2 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.2.2 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.2.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.2.1 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.2.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.2.0 训练 CPU,GPU py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.1.3 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.1.3 训练 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.1.2 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.1.2 训练 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.1.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.1.1 训练 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.1.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.1.0 训练 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.4 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.4 训练 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.3 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.3 训练 CPU,GPU py3、py36
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.2 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.2 训练 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.1 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.1 训练 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.0 训练 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.5 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.5 训练 CPU,GPU py3、py36、py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.4 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.4 训练 CPU,GPU py3、py36、py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.3 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.3 训练 CPU,GPU py2、py3、py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.2 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.2 训练 CPU,GPU py2、py3、py37
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.0 训练 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 1.14.0 推理 CPU,GPU -
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> 1.14.0 训练 CPU,GPU py2、py3
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> 1.13.0 推理 CPU,GPU -