Docker Registry Paths and Example Code for Asia Pacific (Mumbai) (ap-south-1) - ECR Paths

Docker Registry Paths and Example Code for Asia Pacific (Mumbai) (ap-south-1)

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

AutoGluon (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

BlazingText (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Chainer (DLC)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='chainer',region='ap-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
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-chainer:<tag> 5.0.0 inference, training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-chainer:<tag> 4.1.0 inference, training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-chainer:<tag> 4.0.0 inference, training CPU, GPU py2, py3

Clarify (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

DJL DeepSpeed (algorithm)

SageMaker AI 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)
763104351884.dkr.ecr.ap-south-1.amazonaws.com/djl-inference:0.22.1-deepspeed0.8.3-cu118-<tag> 0.22.1 inference
763104351884.dkr.ecr.ap-south-1.amazonaws.com/djl-inference:0.21.0-deepspeed0.8.3-cu117-<tag> 0.21.0 inference
763104351884.dkr.ecr.ap-south-1.amazonaws.com/djl-inference:0.20.0-deepspeed0.7.5-cu116-<tag> 0.20.0 inference
763104351884.dkr.ecr.ap-south-1.amazonaws.com/djl-inference:0.19.0-deepspeed0.7.3-cu113-<tag> 0.19.0 inference

Data Wrangler (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Debugger (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

DeepAR Forecasting (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Factorization Machines (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Hugging Face (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

IP Insights (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Image classification (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Inferentia MXNet (DLC)

SageMaker AI Python SDK example to retrieve registry path.

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

Inferentia PyTorch (DLC)

SageMaker AI Python SDK example to retrieve registry path.

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

K-Means (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

KNN (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

LDA (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Linear Learner (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

MXNet (DLC)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='mxnet',region='ap-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
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-training:<tag> 1.9.0 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-inference:<tag> 1.9.0 inference CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-training:<tag> 1.8.0 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-inference:<tag> 1.8.0 inference CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-training:<tag> 1.7.0 training CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-inference:<tag> 1.7.0 inference CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-inference-eia:<tag> 1.7.0 eia CPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-training:<tag> 1.6.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-inference:<tag> 1.6.0 inference CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-inference-eia:<tag> 1.5.1 eia CPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.4.1 training CPU, GPU py2
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-training:<tag> 1.4.1 training CPU, GPU py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet-serving:<tag> 1.4.1 inference CPU, GPU py2
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-inference:<tag> 1.4.1 inference CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/mxnet-inference-eia:<tag> 1.4.1 eia CPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.4.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet-serving:<tag> 1.4.0 inference CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet-serving-eia:<tag> 1.4.0 eia CPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.3.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.3.0 inference CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet-eia:<tag> 1.3.0 eia CPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.2.1 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.2.1 inference CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.1.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.1.0 inference CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.0.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 1.0.0 inference CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 0.12.1 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-mxnet:<tag> 0.12.1 inference CPU, GPU py2, py3

MXNet Coach (DLC)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='coach-mxnet',region='ap-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
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-mxnet:coach0.11.0-<tag> 0.11.0 training CPU, GPU py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-mxnet:coach0.11-<tag> 0.11 training CPU, GPU py3

Model Monitor (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

NTM (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Neo Image Classification (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Neo MXNet (DLC)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='neo-mxnet',region='ap-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
763008648453.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-inference-mxnet:<tag> 1.8 inference CPU, GPU py3

Neo PyTorch (DLC)

SageMaker AI Python SDK example to retrieve registry path.

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

Neo Tensorflow (DLC)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='neo-tensorflow',region='ap-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
763008648453.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-inference-tensorflow:<tag> 2.9.2 inference CPU, GPU py3
763008648453.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-inference-tensorflow:<tag> 1.15.3 inference CPU, GPU py3

Neo XGBoost (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Object Detection (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Object2Vec (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

PCA (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

PyTorch (DLC)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='pytorch',region='ap-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
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 2.0.0 inference CPU, GPU py310
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference-graviton:<tag> 2.0.0 inference_graviton CPU py310
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 2.0.0 training CPU, GPU py310
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.13.1 inference CPU, GPU py39
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.13.1 training CPU, GPU py39
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.12.1 inference CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference-graviton:<tag> 1.12.1 inference_graviton CPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.12.1 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.12.0 inference CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.12.0 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.11.0 inference CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.11.0 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.10.2 inference CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.10.2 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.10.0 inference CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.10.0 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.9.1 inference CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.9.1 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.9.0 inference CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.9.0 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.8.1 inference CPU, GPU py3, py36
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.8.1 training CPU, GPU py3, py36
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.8.0 inference CPU, GPU py3, py36
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.8.0 training CPU, GPU py3, py36
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.7.1 inference CPU, GPU py3, py36
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.7.1 training CPU, GPU py3, py36
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.6.0 inference CPU, GPU py3, py36
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.6.0 training CPU, GPU py3, py36
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.5.0 inference CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.5.0 training CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.4.0 inference CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.4.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference-eia:<tag> 1.3.1 eia CPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.3.1 inference CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.3.1 training CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-inference:<tag> 1.2.0 inference CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training:<tag> 1.2.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-pytorch:<tag> 1.1.0 inference CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-pytorch:<tag> 1.1.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-pytorch:<tag> 1.0.0 inference CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-pytorch:<tag> 1.0.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-pytorch:<tag> 0.4.0 inference CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-pytorch:<tag> 0.4.0 training CPU, GPU py2, py3

PyTorch Neuron (DLC)

SageMaker AI 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
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-training-neuron:<tag> 1.11.0 training TRN py38

PyTorch Training Compiler (DLC)

SageMaker AI 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
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-trcomp-training:<tag> 1.13.1 training GPU py39
763104351884.dkr.ecr.ap-south-1.amazonaws.com/pytorch-trcomp-training:<tag> 1.12.0 training GPU py38

Random Cut Forest (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Ray PyTorch (DLC)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='ray-pytorch',region='ap-south-1',version='0.8.5',instance_type='ml.c5.4xlarge')
Registry path Version Job types (image scope) Processor types Python versions
462105765813.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-ray-container:ray-1.6.0-torch-<tag> 1.6.0 training CPU, GPU py36
462105765813.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-ray-container:ray-0.8.5-torch-<tag> 0.8.5 training CPU, GPU py36

Scikit-learn (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Semantic Segmentation (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Seq2Seq (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Spark (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='spark',region='ap-south-1',version='3.0',image_scope='processing')
Registry path Version Job types (image scope)
105495057255.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-spark-processing:<tag> 3.2 processing
105495057255.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-spark-processing:<tag> 3.1 processing
105495057255.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-spark-processing:<tag> 3.0 processing
105495057255.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-spark-processing:<tag> 2.4 processing

SparkML Serving (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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

Tensorflow (DLC)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='tensorflow',region='ap-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
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.12.1 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.12.0 training CPU, GPU py310
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.11.1 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.11.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.11.0 training CPU, GPU py39
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.10.1 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.10.1 training CPU, GPU py39
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.10.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.9.3 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.9.2 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.9.2 training CPU, GPU py39
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference-graviton:<tag> 2.9.1 inference_graviton CPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.8.4 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.8.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.8.0 training CPU, GPU py39
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.7.1 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.7.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.6.3 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.6.3 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.6.2 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.6.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.6.0 training CPU, GPU py38
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.5.1 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.5.1 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.5.0 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.4.3 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.4.3 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.4.1 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.4.1 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.3.2 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.3.2 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.3.1 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.3.1 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference-eia:<tag> 2.3.0 eia CPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.3.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.3.0 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.2.2 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.2.2 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.2.1 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.2.1 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.2.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.2.0 training CPU, GPU py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.1.3 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.1.3 training CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.1.2 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.1.2 training CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.1.1 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.1.1 training CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.1.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.1.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.4 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.4 training CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.3 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.3 training CPU, GPU py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.2 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.2 training CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.1 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.1 training CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference-eia:<tag> 2.0.0 eia CPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 2.0.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 2.0.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.5 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.5 training CPU, GPU py3, py36, py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.4 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.4 training CPU, GPU py3, py36, py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.3 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.3 training CPU, GPU py2, py3, py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.2 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.2 training CPU, GPU py2, py3, py37
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference-eia:<tag> 1.15.0 eia CPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 1.15.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 1.15.0 training CPU, GPU py2, py3
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference-eia:<tag> 1.14.0 eia CPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 1.14.0 inference CPU, GPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 1.14.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow-scriptmode:<tag> 1.13.1 training CPU, GPU py2
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-training:<tag> 1.13.1 training CPU, GPU py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow-serving-eia:<tag> 1.13.0 eia CPU -
763104351884.dkr.ecr.ap-south-1.amazonaws.com/tensorflow-inference:<tag> 1.13.0 inference CPU, GPU -
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow-serving-eia:<tag> 1.12.0 eia CPU -
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow-serving:<tag> 1.12.0 inference CPU, GPU -
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow-scriptmode:<tag> 1.12.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow-serving-eia:<tag> 1.11.0 eia CPU -
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow-serving:<tag> 1.11.0 inference CPU, GPU -
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow-scriptmode:<tag> 1.11.0 training CPU, GPU py2, py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow-eia:<tag> 1.10.0 eia CPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.10.0 inference CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.10.0 training CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.9.0 inference CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.9.0 training CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.8.0 inference CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.8.0 training CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.7.0 inference CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.7.0 training CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.6.0 inference CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.6.0 training CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.5.0 inference CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.5.0 training CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.4.1 inference CPU, GPU py2
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-tensorflow:<tag> 1.4.1 training CPU, GPU py2

Tensorflow Coach (DLC)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='coach-tensorflow',region='ap-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
462105765813.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-coach-container:coach-1.0.0-tf-<tag> 1.0.0 training CPU, GPU py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.11.1-<tag> 0.11.1 training CPU, GPU py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.11.0-<tag> 0.11.0 training CPU, GPU py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.11-<tag> 0.11 training CPU, GPU py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.10.1-<tag> 0.10.1 training CPU, GPU py3
520713654638.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-tensorflow:coach0.10-<tag> 0.10 training CPU, GPU py3

Tensorflow Inferentia (DLC)

SageMaker AI Python SDK example to retrieve registry path.

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

Tensorflow Ray (DLC)

SageMaker AI Python SDK example to retrieve registry path.

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

VW (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

from sagemaker import image_uris image_uris.retrieve(framework='vw',region='ap-south-1',version='8.7.0',image_scope='training')
Registry path Version Job types (image scope)
462105765813.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-rl-vw-container:vw-8.7.0-<tag> 8.7.0 training

XGBoost (algorithm)

SageMaker AI Python SDK example to retrieve registry path.

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