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

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

针对美国东部(俄亥俄)(us-east-2) 的 Docker 注册表路径和示例代码

以下主题列出了 Amazon SageMaker 提供的此区域中每个算法和深度学习容器的参数。

BlazingText (算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='blazingtext',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
825641698319.dkr.ecr.us-east-2.amazonaws.com/blazingtext:<tag> 1 推理、推理

查纳 (DLC)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='chainer',region='us-east-2',version='5.0.0',py_version='py3',image_scope='inference',instance_type='ml.c5.4xlarge')
注册表路径 Version Job 类型(图像范围) 处理器类型 Python 版本
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-chainer:<tag> 4.0.0 推理、推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-chainer:<tag> 4.1.0 推理、推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-chainer:<tag> 5.0.0 推理、推理 CPU, GPU Py2, Py3

澄清算法

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='clarify',region='us-east-2',version='1.0',image_scope='processing')
注册表路径 Version Job 类型(图像范围)
211330385671.dkr.ecr.us-east-2.amazonaws.com/sagemaker-clarify-processing<tag> 1.0 处理

数据牧人 (算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='data-wrangler',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
415577184552.dkr.ecr.us-east-2.amazonaws.com/sagemaker-data-wrangler-container<tag> 1.x 处理

调试器 (算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='debugger',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
915447279597.dkr.ecr.us-east-2.amazonaws.com/sagemaker-debugger-rules:<tag> latest Debugger

DeepAR 预测(算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='forecasting-deepar',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
566113047672.dkr.ecr.us-east-2.amazonaws.com/forecasting-deepar:<tag> 1 推理、推理

分解机 (算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='factorization-machines',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
404615174143.dkr.ecr.us-east-2.amazonaws.com/factorization-machines):<tag> 1 推理、推理

拥抱面 (算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='huggingface',region='us-east-2',version='4.4.2',image_scope='training',base_framework_version='tensorflow2.4.1')
注册表路径 Version Job 类型(图像范围)
763104351884.dkr.ecr.us-east-2.amazonaws.com/huggingface-pytorch-training:<tag> 4.4.2 训练
763104351884.dkr.ecr.us-east-2.amazonaws.com/huggingface-pytorch-training:<tag> 4.5.0 训练
763104351884.dkr.ecr.us-east-2.amazonaws.com/huggingface-tensorflow-training:<tag> 4.4.2 训练
763104351884.dkr.ecr.us-east-2.amazonaws.com/huggingface-tensorflow-training:<tag> 4.5.0 训练

IP 见解(算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='ipinsights',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
404615174143.dkr.ecr.us-east-2.amazonaws.com/ipinsights<tag> 1 推理、推理

图像分类(算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='image-classification',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
825641698319.dkr.ecr.us-east-2.amazonaws.com/image-classification<tag> 1 推理、推理

推理 MxNet(DLC)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='inferentia-mxnet',region='us-east-2',version='1.5.1',instance_type='ml.inf1.6xlarge')
注册表路径 Version Job 类型(图像范围) 处理器类型 Python 版本
007439368137.dkr.ecr.us-east-2.amazonaws.com/sagemaker-neo-mxnet:<tag> 1.5.1 推理 inf Py3

PyTorch 推理(DLC)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='inferentia-pytorch',region='us-east-2',version='1.5.1',py_version='py3')
注册表路径 Version Job 类型(图像范围) 处理器类型 Python 版本
007439368137.dkr.ecr.us-east-2.amazonaws.com/sagemaker-neo-pytorch<tag> 1.5.1 推理 in Py

K-Moes 算法

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='kmeans',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
404615174143.dkr.ecr.us-east-2.amazonaws.com/kmeans 日<tag> 1 推理、推理

KNN 算法

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='knn',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
404615174143.dkr.ecr.us-east-2.amazonaws.com/knn<tag> 1 推理、推理

LDA 算法

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='lda',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
999911452149.dkr.ecr.us-east-2.amazonaws.com/lda:<tag> 1 推理、推理

线性学习器 (算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='linear-learner',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
40464615174143 英文版本-东部-2 日博物馆/线性学习器:<tag> 1 推理、推理

MxNet(数据库)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='mxnet',region='us-east-2',version='1.4.1',py_version='py3',image_scope='inference', instance_type='ml.c5.4xlarge')
注册表路径 Version Job 类型(图像范围) 处理器类型 Python 版本
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet-eia:<tag> 1.3.0 EIA CPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet-serving-eia<tag> 1.4.0 EIA CPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet-serving:<tag> 1.4.0 推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet-serving:<tag> 1.4.1 推理 CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 0.12.1 推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 0.12.1 训练 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.0.0 推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.0.0 训练 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.1.0 推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.1.0 训练 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.2.1 推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.2.1 训练 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.3.0 推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.3.0 训练 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.4.0 训练 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-mxnet<tag> 1.4.1 训练 CPU, GPU py2
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-inference-eia<tag> 1.4.1 EIA CPU Py2, Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-inference-eia<tag> 1.5.1 EIA CPU Py2, Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-inference-eia<tag> 1.7.0 EIA CPU Py
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-inference<tag> 1.4.1 推理 CPU, GPU Py
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-inference<tag> 1.6.0 推理 CPU, GPU Py2, Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-inference<tag> 1.7.0 推理 CPU, GPU Py
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-inference<tag> 1.8.0 推理 CPU, GPU Py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-training:<tag> 1.4.1 CPU, GPU Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-training:<tag> 1.6.0 CPU, GPU Py2, Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-training:<tag> 1.7.0 训练 CPU, GPU Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/mxnet-training:<tag> 1.8.0 CPU, GPU Py37

MxNet 教练 (DLC)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='coach-mxnet',region='us-east-2',version='0.11',py_version='py3',image_scope='training',instance_type='ml.c5.4xlarge')
注册表路径 Version Job 类型(图像范围) 处理器类型 Python 版本
东亚马逊/制造商-无线网络:客车 0.11-<tag> 0.11 训练 CPU, GPU Py3
东亚马逊物流/制造商-无线网络:客车 0.11.0-<tag> 0.11.0 训练 CPU, GPU Py3

模型监视器 (算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='model-monitor',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
777275614652.dkr.ecr.us-east-2.amazonaws.com/sagemaker-model-monitor-analyzer orFlow Flow<tag> 监控

NTM 算法

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='ntm',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
404615174143.dkr.ecr.us-east-2.amazonaws.com/ntm:<tag> 1 推理、推理

新图像分类(算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='image-classification-neo',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
007439368137.dkr.ecr.us-east-2.amazonaws.com/image-classification-neo:<tag> latest 推理

新 MXNet (DLC)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='neo-mxnet',region='us-east-2',version='1.8',py_version='py3',image_scope='inference', instance_type='ml.c5.4xlarge')
注册表路径 Version Job 类型(图像范围) 处理器类型 Python 版本
007439368137.dkr.ecr.us-east-2.amazonaws.com/sagemaker-inference-mxnet<tag> 1.8 推理 CPU, GPU Py3

新派 PyTorch (DLC)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='neo-pytorch',region='us-east-2',version='1.6',image_scope='inference',instance_type='ml.c5.4xlarge')
注册表路径 Version Job 类型(图像范围) 处理器类型 Python 版本
007439368137.dkr.ecr.us-east-2.amazonaws.com/sagemaker-inference-pytorch 理<tag> 1.4 推理 CPU, GPU Py3
007439368137.dkr.ecr.us-east-2.amazonaws.com/sagemaker-inference-pytorch 理<tag> 1.5 推理 CPU, GPU Py3
007439368137.dkr.ecr.us-east-2.amazonaws.com/sagemaker-inference-pytorch 理<tag> 1.6 推理 CPU, GPU Py3

新温度流 (DLC)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='neo-tensorflow',region='us-east-2',version='1.15.3',py_version='py3',instance_type='ml.c5.4xlarge')
注册表路径 Version Job 类型(图像范围) 处理器类型 Python 版本
007439368137.dkr.ecr.us-east-2.amazonaws.com/sagemaker-inference-tensorflow:<tag> 1.15.3 推理 CPU, GPU Py3

新 XGBoost(算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='xgboost-neo',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
007439368137.dkr.ecr.us-east-2.amazonaws.com/xgboost-neo:<tag> latest 推理

物体检测 (算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='object-detection',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
825641698319.dkr.ecr.us-east-2.amazonaws.com/object-detection<tag> 1 推理、推理

Object2Vec 算法

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='object2vec',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
404615174143.dkr.ecr.us-east-2.amazonaws.com/object2vec:<tag> 1 推理、推理

PCA (算法)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='pca',region='us-east-2')
注册表路径 Version Job 类型(图像范围)
404615174143.dkr.ecr.us-east-2.amazonaws.com/pca<tag> 1 推理、推理

PyTorch (DLC)

SageMaker Python 软件开发工具包示例来检索注册表路径。

from sagemaker import image_uris image_uris.retrieve(framework='pytorch',region='us-east-2',version='1.8.0',py_version='py3',image_scope='inference', instance_type='ml.c5.4xlarge')
注册表路径 Version Job 类型(图像范围) 处理器类型 Python 版本
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-pytorch 理<tag> 0.4.0 推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-pytorch 理<tag> 0.4.0 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-pytorch 理<tag> 1.0.0 推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-pytorch 理<tag> 1.0.0 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-pytorch 理<tag> 1.1.0 推理 CPU, GPU Py2, Py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-pytorch 理<tag> 1.1.0 训练 CPU, GPU Py2, Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-inference-eia<tag> 1.3.1 EIA CPU Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-inference<tag> 1.2.0 推理 CPU, GPU Py2, Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-inference<tag> 1.3.1 推理 CPU, GPU Py2, Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-inference<tag> 1.4.0 推理 CPU, GPU Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-inference<tag> 1.5.0 推理 CPU, GPU Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-inference<tag> 1.6.0 推理 CPU, GPU 三、三、三、三
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-inference<tag> 1.7.1 推理 CPU, GPU 三、三、三、三
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-inference<tag> 1.8.0 推理 CPU, GPU 三、三、三、三
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-inference<tag> 1.8.1 推理 CPU, GPU 三、三、三、三
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-training<tag> 1.2.0 训练 CPU, GPU Py2, Py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-training<tag> 1.3.1 training CPU, GPU py2, py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-training:<tag> 1.4.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-training:<tag> 1.5.0 training CPU, GPU py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-training:<tag> 1.6.0 training CPU, GPU py3, py36
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-training:<tag> 1.7.1 training CPU, GPU py3, py36
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-training:<tag> 1.8.0 training CPU, GPU py3, py36
763104351884.dkr.ecr.us-east-2.amazonaws.com/pytorch-training:<tag> 1.8.1 training CPU, GPU py3, py36

Random Cut Forest (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='randomcutforest',region='us-east-2')
注册表路径 Version Job types (image scope)
404615174143.dkr.ecr.us-east-2.amazonaws.com/randomcutforest:<tag> 1 inference, training

Ray PyTorch (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='ray-pytorch',region='us-east-2',version='0.8.5',instance_type='ml.c5.4xlarge')
注册表路径 Version Job types (image scope) Processor types Python versions
462105765813.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-ray-container:ray-0.8.5-torch-<tag> 0.8.5 training CPU, GPU py36

Scikit-learn (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='sklearn',region='us-east-2',version='0.23-1',image_scope='inference')
注册表路径 Version Job types (image scope)
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-scikit-learn:<tag> 0.20.0 inference, training
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-scikit-learn:<tag> 0.23-1 inference, training

Semantic Segmentation (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='semantic-segmentation',region='us-east-2')
注册表路径 Version Job types (image scope)
825641698319.dkr.ecr.us-east-2.amazonaws.com/semantic-segmentation:<tag> 1 inference, training

Seq2Seq (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='seq2seq',region='us-east-2')
注册表路径 Version Job types (image scope)
825641698319.dkr.ecr.us-east-2.amazonaws.com/seq2seq:<tag> 1 inference, training

Spark (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='spark',region='us-east-2',version='3.0',image_scope='processing')
注册表路径 Version Job types (image scope)
314815235551.dkr.ecr.us-east-2.amazonaws.com/sagemaker-spark-processing:<tag> 2.4 processing
314815235551.dkr.ecr.us-east-2.amazonaws.com/sagemaker-spark-processing:<tag> 3.0 processing

SparkML Serving (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='sparkml-serving',region='us-east-2',version='2.4')
注册表路径 Version Job types (image scope)
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-sparkml-serving:<tag> 2.2 inference
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-sparkml-serving:<tag> 2.4 inference

Tensorflow (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='tensorflow',region='us-east-2',version='1.12.0',image_scope='inference',instance_type='ml.c5.4xlarge')
注册表路径 Version Job types (image scope) Processor types Python versions
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow-eia:<tag> 1.10.0 eia CPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow-scriptmode:<tag> 1.11.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow-scriptmode:<tag> 1.12.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow-scriptmode:<tag> 1.13.1 training CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow-serving-eia:<tag> 1.11.0 eia CPU -
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow-serving-eia:<tag> 1.12.0 eia CPU -
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow-serving-eia:<tag> 1.13.0 eia CPU -
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow-serving:<tag> 1.11.0 inference CPU, GPU -
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow-serving:<tag> 1.12.0 inference CPU, GPU -
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.10.0 inference CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.10.0 training CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.4.1 inference CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.4.1 training CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.5.0 inference CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.5.0 training CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.6.0 inference CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.6.0 training CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.7.0 inference CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.7.0 training CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.8.0 inference CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.8.0 training CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.9.0 inference CPU, GPU py2
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-tensorflow:<tag> 1.9.0 training CPU, GPU py2
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference-eia:<tag> 1.14.0 eia CPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference-eia:<tag> 1.15.0 eia CPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference-eia:<tag> 2.0.0 eia CPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference-eia:<tag> 2.3.0 eia CPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 1.13.0 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 1.14.0 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.0 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.2 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.3 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.4 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 1.15.5 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.0 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.1 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.2 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.3 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.0.4 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.1.0 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.1.1 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.1.2 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.1.3 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.2.0 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.2.1 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.2.2 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.3.0 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.3.1 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.3.2 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-inference:<tag> 2.4.1 inference CPU, GPU -
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 1.13.1 training CPU, GPU py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 1.14.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.2 training CPU, GPU py2, py3, py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.3 training CPU, GPU py2, py3, py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.4 training CPU, GPU py3, py36, py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 1.15.5 training CPU, GPU py3, py36, py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.1 training CPU, GPU py2, py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.2 training CPU, GPU py2, py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.3 training CPU, GPU py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.0.4 training CPU, GPU py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.1.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.1.1 training CPU, GPU py2, py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.1.2 training CPU, GPU py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.1.3 training CPU, GPU py3
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.2.0 training CPU, GPU py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.2.1 training CPU, GPU py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.2.2 training CPU, GPU py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.3.0 training CPU, GPU py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.3.1 training CPU, GPU py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.3.2 training CPU, GPU py37
763104351884.dkr.ecr.us-east-2.amazonaws.com/tensorflow-training:<tag> 2.4.1 training CPU, GPU py37

Tensorflow Coach (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='coach-tensorflow',region='us-east-2',version='1.0.0',image_scope='training',instance_type='ml.c5.4xlarge')
注册表路径 Version Job types (image scope) Processor types Python versions
462105765813.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-coach-container:coach-1.0.0-tf-<tag> 1.0.0 training CPU, GPU py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-tensorflow:coach0.10-<tag> 0.10 training CPU, GPU py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-tensorflow:coach0.10.1-<tag> 0.10.1 training CPU, GPU py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-tensorflow:coach0.11-<tag> 0.11 training CPU, GPU py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-tensorflow:coach0.11.0-<tag> 0.11.0 training CPU, GPU py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-tensorflow:coach0.11.1-<tag> 0.11.1 training CPU, GPU py3

Tensorflow Inferentia (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='inferentia-tensorflow',region='us-east-2',version='1.15.0',instance_type='ml.inf1.6xlarge')
注册表路径 Version Job types (image scope) Processor types Python versions
007439368137.dkr.ecr.us-east-2.amazonaws.com/sagemaker-neo-tensorflow:<tag> 1.15.0 inference inf py3

Tensorflow Ray (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='ray-tensorflow',region='us-east-2',version='0.8.5',instance_type='ml.c5.4xlarge')
注册表路径 Version Job types (image scope) Processor types Python versions
462105765813.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-ray-container:ray-0.8.2-tf-<tag> 0.8.2 training CPU, GPU py36
462105765813.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-ray-container:ray-0.8.5-tf-<tag> 0.8.5 training CPU, GPU py36
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-tensorflow:ray0.5-<tag> 0.5 training CPU, GPU py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-tensorflow:ray0.5.3-<tag> 0.5.3 training CPU, GPU py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-tensorflow:ray0.6-<tag> 0.6 training CPU, GPU py3
520713654638.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-tensorflow:ray0.6.5-<tag> 0.6.5 training CPU, GPU py3

VW (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='vw',region='us-east-2',version='8.7.0',image_scope='training')
注册表路径 Version Job types (image scope)
462105765813.dkr.ecr.us-east-2.amazonaws.com/sagemaker-rl-vw-container:vw-8.7.0-<tag> 8.7.0 training

XGBoost (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='xgboost',region='us-east-2',version='1.2-1')
注册表路径 Version Job types (image scope)
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-xgboost:<tag> 0.90-1 inference, training
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-xgboost:<tag> 0.90-2 inference, training
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-xgboost:<tag> 1.0-1 inference, training
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-xgboost:<tag> 1.2-1 inference, training
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-xgboost:<tag> 1.2-2 inference, training
257758044811.dkr.ecr.us-east-2.amazonaws.com/sagemaker-xgboost:<tag> 1.3-1 inference, training
825641698319.dkr.ecr.us-east-2.amazonaws.com/xgboost:<tag> 1 inference, training