Docker Registry Paths and Example Code for Middle East (Bahrain) (me-south-1) - Amazon SageMaker ECR Paths
Services or capabilities described in Amazon Web Services documentation might vary by Region. To see the differences applicable to the China Regions, see Getting Started with Amazon Web Services in China (PDF).

Docker Registry Paths and Example Code for Middle East (Bahrain) (me-south-1)

The following topics list parameters for each of the algorithms and deep learning containers in this region provided by Amazon SageMaker.

AutoGluon (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='autogluon',region='me-south-1',image_scope='inference',version='0.4')
Registry path Version Job types (image scope)
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-training:<tag> 0.7.0 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-inference:<tag> 0.7.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-training:<tag> 0.6.2 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-inference:<tag> 0.6.2 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-training:<tag> 0.6.1 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-inference:<tag> 0.6.1 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-training:<tag> 0.5.2 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-inference:<tag> 0.5.2 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-training:<tag> 0.4.3 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-inference:<tag> 0.4.3 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-training:<tag> 0.4.2 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-inference:<tag> 0.4.2 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-training:<tag> 0.4.0 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-inference:<tag> 0.4.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-training:<tag> 0.3.2 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-inference:<tag> 0.3.2 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-training:<tag> 0.3.1 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/autogluon-inference:<tag> 0.3.1 inference

BlazingText (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='blazingtext',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/blazingtext:<tag> 1 inference, training

Chainer (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='chainer',region='me-south-1',version='5.0.0',py_version='py3',image_scope='inference',instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-chainer:<tag> 5.0.0 inference, training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-chainer:<tag> 4.1.0 inference, training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-chainer:<tag> 4.0.0 inference, training CPU, GPU py2, py3

Clarify (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='clarify',region='me-south-1',version='1.0',image_scope='processing')
Registry path Version Job types (image scope)
835444307964.dkr.ecr.me-south-1.amazonaws.com/sagemaker-clarify-processing:<tag> 1.0 processing

DJL DeepSpeed (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='djl-deepspeed', region='us-west-2',py_version='py3',image_scope='inference')
Registry path Version Job types (image scope)
217643126080.dkr.ecr.me-south-1.amazonaws.com/djl-inference:0.22.1-deepspeed0.8.3-cu118-<tag> 0.22.1 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/djl-inference:0.21.0-deepspeed0.8.3-cu117-<tag> 0.21.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/djl-inference:0.20.0-deepspeed0.7.5-cu116-<tag> 0.20.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/djl-inference:0.19.0-deepspeed0.7.3-cu113-<tag> 0.19.0 inference

Data Wrangler (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='data-wrangler',region='me-south-1')
Registry path Version Job types (image scope)
376037874950.dkr.ecr.me-south-1.amazonaws.com/sagemaker-data-wrangler-container:<tag> 2.x processing
376037874950.dkr.ecr.me-south-1.amazonaws.com/sagemaker-data-wrangler-container:<tag> 1.x processing

Debugger (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='debugger',region='me-south-1')
Registry path Version Job types (image scope)
986000313247.dkr.ecr.me-south-1.amazonaws.com/sagemaker-debugger-rules:<tag> latest debugger

DeepAR Forecasting (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='forecasting-deepar',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/forecasting-deepar:<tag> 1 inference, training

Factorization Machines (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='factorization-machines',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/factorization-machines:<tag> 1 inference, training

Hugging Face (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='huggingface',region='me-south-1',version='4.4.2',image_scope='training',base_framework_version='tensorflow2.4.1')
Registry path Version Job types (image scope)
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.26.0 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-inference:<tag> 4.26.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-inference:<tag> 4.26.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.17.0 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-training:<tag> 4.17.0 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-inference:<tag> 4.17.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-inference:<tag> 4.17.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.12.3 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-training:<tag> 4.12.3 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-inference:<tag> 4.12.3 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-inference:<tag> 4.12.3 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.11.0 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-training:<tag> 4.11.0 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-inference:<tag> 4.11.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-inference:<tag> 4.11.0 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.10.2 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.10.2 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-training:<tag> 4.10.2 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-training:<tag> 4.10.2 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-inference:<tag> 4.10.2 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-inference:<tag> 4.10.2 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-inference:<tag> 4.10.2 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-inference:<tag> 4.10.2 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.6.1 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.6.1 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.6.1 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-training:<tag> 4.6.1 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-inference:<tag> 4.6.1 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-inference:<tag> 4.6.1 inference
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.5.0 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-training:<tag> 4.5.0 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-pytorch-training:<tag> 4.4.2 training
217643126080.dkr.ecr.me-south-1.amazonaws.com/huggingface-tensorflow-training:<tag> 4.4.2 training

IP Insights (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='ipinsights',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/ipinsights:<tag> 1 inference, training

Image classification (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='image-classification',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/image-classification:<tag> 1 inference, training

Inferentia MXNet (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='inferentia-mxnet',region='me-south-1',version='1.5.1',instance_type='ml.inf1.6xlarge')
Registry path Version Job types (image scope) Processor types Python versions
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-neo-mxnet:<tag> 1.8 inference inf py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-neo-mxnet:<tag> 1.5.1 inference inf py3

Inferentia PyTorch (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='inferentia-pytorch',region='me-south-1',version='1.9',py_version='py3')
Registry path Version Job types (image scope) Processor types Python versions
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-neo-pytorch:<tag> 1.9 inference inf py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-neo-pytorch:<tag> 1.8 inference inf py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-neo-pytorch:<tag> 1.7 inference inf py3

K-Means (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='kmeans',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/kmeans:<tag> 1 inference, training

KNN (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='knn',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/knn:<tag> 1 inference, training

Linear Learner (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='linear-learner',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/linear-learner:<tag> 1 inference, training

MXNet (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='mxnet',region='me-south-1',version='1.4.1',py_version='py3',image_scope='inference', instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-training:<tag> 1.9.0 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-inference:<tag> 1.9.0 inference CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-training:<tag> 1.8.0 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-inference:<tag> 1.8.0 inference CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-training:<tag> 1.7.0 training CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-inference:<tag> 1.7.0 inference CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-inference-eia:<tag> 1.7.0 eia CPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-training:<tag> 1.6.0 training CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-inference:<tag> 1.6.0 inference CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-inference-eia:<tag> 1.5.1 eia CPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.4.1 training CPU, GPU py2
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-training:<tag> 1.4.1 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet-serving:<tag> 1.4.1 inference CPU, GPU py2
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-inference:<tag> 1.4.1 inference CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/mxnet-inference-eia:<tag> 1.4.1 eia CPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.4.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet-serving:<tag> 1.4.0 inference CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet-serving-eia:<tag> 1.4.0 eia CPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.3.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.3.0 inference CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet-eia:<tag> 1.3.0 eia CPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.2.1 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.2.1 inference CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.1.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.1.0 inference CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.0.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.0.0 inference CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 0.12.1 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-mxnet:<tag> 0.12.1 inference CPU, GPU py2, py3

MXNet Coach (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='coach-mxnet',region='me-south-1',version='0.11',py_version='py3',image_scope='training',instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-mxnet:coach0.11.0-<tag> 0.11.0 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-mxnet:coach0.11-<tag> 0.11 training CPU, GPU py3

Model Monitor (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='model-monitor',region='me-south-1')
Registry path Version Job types (image scope)
607024016150.dkr.ecr.me-south-1.amazonaws.com/sagemaker-model-monitor-analyzer:<tag> monitoring

NTM (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='ntm',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/ntm:<tag> 1 inference, training

Neo Image Classification (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='image-classification-neo',region='me-south-1')
Registry path Version Job types (image scope)
836785723513.dkr.ecr.me-south-1.amazonaws.com/image-classification-neo:<tag> latest inference

Neo MXNet (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='neo-mxnet',region='me-south-1',version='1.8',py_version='py3',image_scope='inference', instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-mxnet:<tag> 1.8 inference CPU, GPU py3

Neo PyTorch (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='neo-pytorch',region='me-south-1',version='1.6',image_scope='inference',instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-pytorch:<tag> 1.13 inference CPU, GPU py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-pytorch:<tag> 1.12 inference CPU, GPU py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-pytorch:<tag> 1.8 inference CPU, GPU py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-pytorch:<tag> 1.7 inference CPU, GPU py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-pytorch:<tag> 1.6 inference CPU, GPU py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-pytorch:<tag> 1.5 inference CPU, GPU py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-pytorch:<tag> 1.4 inference CPU, GPU py3

Neo Tensorflow (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='neo-tensorflow',region='me-south-1',version='1.15.3',py_version='py3',instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-tensorflow:<tag> 2.9.2 inference CPU, GPU py3
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-inference-tensorflow:<tag> 1.15.3 inference CPU, GPU py3

Neo XGBoost (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='xgboost-neo',region='me-south-1')
Registry path Version Job types (image scope)
836785723513.dkr.ecr.me-south-1.amazonaws.com/xgboost-neo:<tag> latest inference

Object Detection (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='object-detection',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/object-detection:<tag> 1 inference, training

Object2Vec (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='object2vec',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/object2vec:<tag> 1 inference, training

PCA (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='pca',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/pca:<tag> 1 inference, training

PyTorch (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='pytorch',region='me-south-1',version='1.8.0',py_version='py3',image_scope='inference', instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 2.0.0 inference CPU, GPU py310
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference-graviton:<tag> 2.0.0 inference_graviton CPU py310
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 2.0.0 training CPU, GPU py310
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.13.1 inference CPU, GPU py39
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.13.1 training CPU, GPU py39
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.12.1 inference CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference-graviton:<tag> 1.12.1 inference_graviton CPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.12.1 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.12.0 inference CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.12.0 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.11.0 inference CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.11.0 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.10.2 inference CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.10.2 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.10.0 inference CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.10.0 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.9.1 inference CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.9.1 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.9.0 inference CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.9.0 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.8.1 inference CPU, GPU py3, py36
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.8.1 training CPU, GPU py3, py36
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.8.0 inference CPU, GPU py3, py36
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.8.0 training CPU, GPU py3, py36
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.7.1 inference CPU, GPU py3, py36
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.7.1 training CPU, GPU py3, py36
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.6.0 inference CPU, GPU py3, py36
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.6.0 training CPU, GPU py3, py36
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.5.0 inference CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.5.0 training CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.4.0 inference CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.4.0 training CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.3.1 inference CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.3.1 training CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-inference:<tag> 1.2.0 inference CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training:<tag> 1.2.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-pytorch:<tag> 1.1.0 inference CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-pytorch:<tag> 1.1.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-pytorch:<tag> 1.0.0 inference CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-pytorch:<tag> 1.0.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-pytorch:<tag> 0.4.0 inference CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-pytorch:<tag> 0.4.0 training CPU, GPU py2, py3

PyTorch Neuron (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='pytorch-neuron',region='us-west-2', image_scope='inference')
Registry path Version Job types (image scope) Processor types Python versions
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-training-neuron:<tag> 1.11.0 training TRN py38

PyTorch Training Compiler (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='pytorch-training-compiler',region='us-west-2', version='py38')
Registry path Version Job types (image scope) Processor types Python versions
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-trcomp-training:<tag> 1.13.1 training GPU py39
217643126080.dkr.ecr.me-south-1.amazonaws.com/pytorch-trcomp-training:<tag> 1.12.0 training GPU py38

Random Cut Forest (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='randomcutforest',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.amazonaws.com/randomcutforest:<tag> 1 inference, training

Scikit-learn (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='sklearn',region='me-south-1',version='0.23-1',image_scope='inference')
Registry path Version Package version Job types (image scope)
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-scikit-learn:<tag> 1.2-1 1.2.1 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-scikit-learn:<tag> 1.2-1 1.2.1 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-scikit-learn:<tag> 1.0-1 1.0.2 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-scikit-learn:<tag> 1.0-1 1.0.2 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-scikit-learn:<tag> 1.0-1 1.0.2 inference_graviton
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-scikit-learn:<tag> 0.23-1 0.23.2 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-scikit-learn:<tag> 0.23-1 0.23.2 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-scikit-learn:<tag> 0.20.0 0.20.0 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-scikit-learn:<tag> 0.20.0 0.20.0 training

Semantic Segmentation (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='semantic-segmentation',region='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.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='me-south-1')
Registry path Version Job types (image scope)
249704162688.dkr.ecr.me-south-1.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='me-south-1',version='3.0',image_scope='processing')
Registry path Version Job types (image scope)
750251592176.dkr.ecr.me-south-1.amazonaws.com/sagemaker-spark-processing:<tag> 3.2 processing
750251592176.dkr.ecr.me-south-1.amazonaws.com/sagemaker-spark-processing:<tag> 3.1 processing
750251592176.dkr.ecr.me-south-1.amazonaws.com/sagemaker-spark-processing:<tag> 3.0 processing
750251592176.dkr.ecr.me-south-1.amazonaws.com/sagemaker-spark-processing:<tag> 2.4 processing

SparkML Serving (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='sparkml-serving',region='me-south-1',version='2.4')
Registry path Version Job types (image scope)
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-sparkml-serving:<tag> 3.3 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-sparkml-serving:<tag> 2.4 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-sparkml-serving:<tag> 2.2 inference

Tensorflow (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='tensorflow',region='me-south-1',version='1.12.0',image_scope='inference',instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.12.1 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.12.0 training CPU, GPU py310
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.11.1 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.11.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.11.0 training CPU, GPU py39
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.10.1 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.10.1 training CPU, GPU py39
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.10.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.9.3 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.9.2 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.9.2 training CPU, GPU py39
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference-graviton:<tag> 2.9.1 inference_graviton CPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.8.4 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.8.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.8.0 training CPU, GPU py39
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.7.1 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.7.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.6.3 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.6.3 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.6.2 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.6.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.6.0 training CPU, GPU py38
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.5.1 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.5.1 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.5.0 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.4.3 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.4.3 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.4.1 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.4.1 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.3.2 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.3.2 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.3.1 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.3.1 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference-eia:<tag> 2.3.0 eia CPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.3.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.3.0 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.2.2 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.2.2 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.2.1 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.2.1 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.2.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.2.0 training CPU, GPU py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.1.3 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.1.3 training CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.1.2 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.1.2 training CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.1.1 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.1.1 training CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.1.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.1.0 training CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.4 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.4 training CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.3 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.3 training CPU, GPU py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.2 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.2 training CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.1 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.1 training CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference-eia:<tag> 2.0.0 eia CPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.0 training CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.5 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.5 training CPU, GPU py3, py36, py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.4 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.4 training CPU, GPU py3, py36, py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.3 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.3 training CPU, GPU py2, py3, py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.2 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.2 training CPU, GPU py2, py3, py37
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference-eia:<tag> 1.15.0 eia CPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.0 training CPU, GPU py2, py3
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference-eia:<tag> 1.14.0 eia CPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 1.14.0 inference CPU, GPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 1.14.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow-scriptmode:<tag> 1.13.1 training CPU, GPU py2
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-training:<tag> 1.13.1 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow-serving-eia:<tag> 1.13.0 eia CPU -
217643126080.dkr.ecr.me-south-1.amazonaws.com/tensorflow-inference:<tag> 1.13.0 inference CPU, GPU -
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow-serving-eia:<tag> 1.12.0 eia CPU -
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow-serving:<tag> 1.12.0 inference CPU, GPU -
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow-scriptmode:<tag> 1.12.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow-serving-eia:<tag> 1.11.0 eia CPU -
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow-serving:<tag> 1.11.0 inference CPU, GPU -
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow-scriptmode:<tag> 1.11.0 training CPU, GPU py2, py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow-eia:<tag> 1.10.0 eia CPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.10.0 inference CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.10.0 training CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.9.0 inference CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.9.0 training CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.8.0 inference CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.8.0 training CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.7.0 inference CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.7.0 training CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.6.0 inference CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.6.0 training CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.5.0 inference CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.5.0 training CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.4.1 inference CPU, GPU py2
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.4.1 training CPU, GPU py2

Tensorflow Coach (DLC)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='coach-tensorflow',region='me-south-1',version='1.0.0',image_scope='training',instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.11.1-<tag> 0.11.1 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.11.0-<tag> 0.11.0 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.11-<tag> 0.11 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.10.1-<tag> 0.10.1 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.10-<tag> 0.10 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='me-south-1',version='1.15.0',instance_type='ml.inf1.6xlarge')
Registry path Version Job types (image scope) Processor types Python versions
836785723513.dkr.ecr.me-south-1.amazonaws.com/sagemaker-neo-tensorflow:<tag> 2.5.2 inference inf py3
836785723513.dkr.ecr.me-south-1.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='me-south-1',version='0.8.5',instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-tensorflow:ray0.6.5-<tag> 0.6.5 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-tensorflow:ray0.6-<tag> 0.6 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-tensorflow:ray0.5.3-<tag> 0.5.3 training CPU, GPU py3
724002660598.dkr.ecr.me-south-1.amazonaws.com/sagemaker-rl-tensorflow:ray0.5-<tag> 0.5 training CPU, GPU py3

XGBoost (algorithm)

SageMaker Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='xgboost',region='me-south-1',version='1.5-1')
Registry path Version Package version Job types (image scope)
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.7-1 1.7.4 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.7-1 1.7.4 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.5-1 1.5.2 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.5-1 1.5.2 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.5-1 1.5.2 inference_graviton
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.3-1 1.3.3 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.3-1 1.3.3 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.3-1 1.3.3 inference_graviton
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.2-2 1.2.0 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.2-2 1.2.0 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.2-1 1.2.0 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.2-1 1.2.0 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.0-1 1.0.0 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 1.0-1 1.0.0 training
249704162688.dkr.ecr.me-south-1.amazonaws.com/xgboost:<tag> 1 0.72 inference
249704162688.dkr.ecr.me-south-1.amazonaws.com/xgboost:<tag> 1 0.72 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 0.90-2 0.90 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 0.90-2 0.90 training
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 0.90-1 0.90 inference
801668240914.dkr.ecr.me-south-1.amazonaws.com/sagemaker-xgboost:<tag> 0.90-1 0.90 training