Docker Registry Paths and Example Code for (ap-east-2)
The following topics list parameters for each of the algorithms and deep learning containers that are provided by Amazon SageMaker AI in this Amazon Web Services Region.
PyTorch (DLC)
For information about the supported and unsupported PyTorch versions, see the Framework Support Policy Table in the Amazon Deep Learning Containers Developer Guide.
The following SageMaker AI Python SDK example shows how to retrieve a specific registry path.
from sagemaker import image_uris image_uris.retrieve(framework='pytorch',region='ap-east-2',version='1.8.0',py_version='py3',image_scope='inference', instance_type='ml.c5.4xlarge')
| Registry path | Version | Job types (image scope) | Processor types | Python versions |
|---|---|---|---|---|
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
2.7.1 | training | CPU, GPU | py312 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
2.6.0 | inference | CPU, GPU | py312 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
2.6.0 | training | CPU, GPU | py312 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
2.5.1 | inference | CPU, GPU | py311 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
2.5.1 | training | CPU, GPU | py311 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
2.4.0 | inference | CPU, GPU | py311 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> |
2.4.0 | inference_graviton | CPU | py311 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
2.4.0 | training | CPU, GPU | py311 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
2.3.0 | inference | CPU, GPU | py311 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> |
2.3.0 | inference_graviton | CPU | py311 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
2.3.0 | training | CPU, GPU | py311 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> |
2.2.1 | inference_graviton | CPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
2.2.0 | inference | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
2.2.0 | training | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
2.1.0 | inference | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> |
2.1.0 | inference_graviton | CPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
2.1.0 | training | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
2.0.1 | inference | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> |
2.0.1 | inference_graviton | CPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
2.0.1 | training | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
2.0.0 | inference | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> |
2.0.0 | inference_graviton | CPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
2.0.0 | training | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.13.1 | inference | CPU, GPU | py39 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.13.1 | training | CPU, GPU | py39 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.12.1 | inference | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference-graviton:<tag> |
1.12.1 | inference_graviton | CPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.12.1 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.12.0 | inference | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.12.0 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.11.0 | inference | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.11.0 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.10.2 | inference | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.10.2 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.10.0 | inference | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.10.0 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.9.1 | inference | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.9.1 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.9.0 | inference | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.9.0 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.8.1 | inference | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.8.1 | training | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.8.0 | inference | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.8.0 | training | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.7.1 | inference | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.7.1 | training | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.6.0 | inference | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.6.0 | training | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.5.0 | inference | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.5.0 | training | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.4.0 | inference | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.4.0 | training | CPU, GPU | py2, py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.3.1 | inference | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.3.1 | training | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-inference:<tag> |
1.2.0 | inference | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/pytorch-training:<tag> |
1.2.0 | training | CPU, GPU | py2, py3 |
Spark (algorithm)
The following SageMaker AI Python SDK example shows how to retrieve a specific registry path.
from sagemaker import image_uris image_uris.retrieve(framework='spark',region='ap-east-2',version='3.0',image_scope='processing')
| Registry path | Version | Job types (image scope) |
|---|---|---|
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> |
3.3 | processing |
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> |
3.2 | processing |
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> |
3.1 | processing |
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> |
3.0 | processing |
533267296287.dkr.ecr.ap-east-2.amazonaws.com/sagemaker-spark-processing:<tag> |
2.4 | processing |
Tensorflow (DLC)
For information about the supported and unsupported TensorFlow versions, see the Framework Support Policy Table in the Amazon Deep Learning Containers Developer Guide.
The following SageMaker AI Python SDK example shows how to retrieve a specific registry path.
from sagemaker import image_uris image_uris.retrieve(framework='tensorflow',region='ap-east-2',version='1.12.0',image_scope='inference',instance_type='ml.c5.4xlarge')
| Registry path | Version | Job types (image scope) | Processor types | Python versions |
|---|---|---|---|---|
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.19.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.19.0 | training | CPU, GPU | py312 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.18.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.18.0 | training | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.16.2 | training | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.16.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> |
2.16.1 | inference_graviton | CPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.14.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> |
2.14.1 | inference_graviton | CPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.14.1 | training | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.13.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> |
2.13.0 | inference_graviton | CPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.13.0 | training | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.12.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> |
2.12.1 | inference_graviton | CPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.12.0 | training | CPU, GPU | py310 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.11.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.11.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.11.0 | training | CPU, GPU | py39 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.10.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.10.1 | training | CPU, GPU | py39 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.10.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.9.3 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.9.2 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.9.2 | training | CPU, GPU | py39 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference-graviton:<tag> |
2.9.1 | inference_graviton | CPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.8.4 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.8.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.8.0 | training | CPU, GPU | py39 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.7.1 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.7.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.6.3 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.6.3 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.6.2 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.6.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.6.0 | training | CPU, GPU | py38 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.5.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.5.1 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.5.0 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.4.3 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.4.3 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.4.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.4.1 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.3.2 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.3.2 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.3.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.3.1 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.3.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.3.0 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.2.2 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.2.2 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.2.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.2.1 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.2.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.2.0 | training | CPU, GPU | py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.1.3 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.1.3 | training | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.1.2 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.1.2 | training | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.1.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.1.1 | training | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.1.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.1.0 | training | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.0.4 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.0.4 | training | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.0.3 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.0.3 | training | CPU, GPU | py3, py36 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.0.2 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.0.2 | training | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.0.1 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.0.1 | training | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
2.0.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
2.0.0 | training | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
1.15.5 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
1.15.5 | training | CPU, GPU | py3, py36, py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
1.15.4 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
1.15.4 | training | CPU, GPU | py3, py36, py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
1.15.3 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
1.15.3 | training | CPU, GPU | py2, py3, py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
1.15.2 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
1.15.2 | training | CPU, GPU | py2, py3, py37 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
1.15.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
1.15.0 | training | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
1.14.0 | inference | CPU, GPU | - |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-training:<tag> |
1.14.0 | training | CPU, GPU | py2, py3 |
975050140332.dkr.ecr.ap-east-2.amazonaws.com/tensorflow-inference:<tag> |
1.13.0 | inference | CPU, GPU | - |