Available Amazon SageMaker Images - Amazon SageMaker
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).

Available Amazon SageMaker Images

Important

As of November 30, 2023, the previous Amazon SageMaker Studio experience is now named Amazon SageMaker Studio Classic. The following section is specific to using the Studio Classic application. For information about using the updated Studio experience, see Amazon SageMaker Studio.

This page lists the SageMaker images and associated kernels that are available in Amazon SageMaker Studio Classic, as well as the format needed to create the ARN for each image. SageMaker images contain the latest Amazon SageMaker Python SDK and the latest version of the kernel. For more information, see Deep Learning Containers Images.

Image ARN format

The following table lists the image ARN and URI format for each Region. To create the full ARN for an image, replace the resource-identifier placeholder with the corresponding resource identifier for the image from the SageMaker images and kernels table. To create the full URI for an image, replace the tag placeholder with the corresponding cpu or gpu tag. For the list of tags you can use, see Supported URI tags.

Note

SageMaker Distribution images use a distinct set of image ARNs, which are listed in the following table.

Region Image ARN Format SageMaker Distribution Image ARN Format SageMaker Distribution Image URI Format
us-east-1 arn:aws:sagemaker:us-east-1:081325390199:image/resource-identifier arn:aws:sagemaker:us-east-1:885854791233:image/resource-identifier 885854791233.dkr.ecr.us-east-1.amazonaws.com/sagemaker-distribution-prod:tag
us-east-2 arn:aws:sagemaker:us-east-2:429704687514:image/resource-identifier arn:aws:sagemaker:us-east-2:137914896644:image/resource-identifier 137914896644.dkr.ecr.us-east-2.amazonaws.com/sagemaker-distribution-prod:tag
us-west-1 arn:aws:sagemaker:us-west-1:742091327244:image/resource-identifier arn:aws:sagemaker:us-west-1:053634841547:image/resource-identifier 053634841547.dkr.ecr.us-west-1.amazonaws.com/sagemaker-distribution-prod:tag
us-west-2 arn:aws:sagemaker:us-west-2:236514542706:image/resource-identifier arn:aws:sagemaker:us-west-2:542918446943:image/resource-identifier 542918446943.dkr.ecr.us-west-2.amazonaws.com/sagemaker-distribution-prod:tag
af-south-1 arn:aws:sagemaker:af-south-1:559312083959:image/resource-identifier arn:aws:sagemaker:af-south-1:238384257742:image/resource-identifier 238384257742.dkr.ecr.af-south-1.amazonaws.com/sagemaker-distribution-prod:tag
ap-east-1 arn:aws:sagemaker:ap-east-1:493642496378:image/resource-identifier arn:aws:sagemaker:ap-east-1:523751269255:image/resource-identifier 523751269255.dkr.ecr.ap-east-1.amazonaws.com/sagemaker-distribution-prod:tag
ap-south-1 arn:aws:sagemaker:ap-south-1:394103062818:image/resource-identifier arn:aws:sagemaker:ap-south-1:245090515133:image/resource-identifier 245090515133.dkr.ecr.ap-south-1.amazonaws.com/sagemaker-distribution-prod:tag
ap-northeast-2 arn:aws:sagemaker:ap-northeast-2:806072073708:image/resource-identifier arn:aws:sagemaker:ap-northeast-2:064688005998:image/resource-identifier 064688005998.dkr.ecr.ap-northeast-2.amazonaws.com/sagemaker-distribution-prod:tag
ap-southeast-1 arn:aws:sagemaker:ap-southeast-1:492261229750:image/resource-identifier arn:aws:sagemaker:ap-southeast-1:022667117163:image/resource-identifier 022667117163.dkr.ecr.ap-southeast-1.amazonaws.com/sagemaker-distribution-prod:tag
ap-southeast-2 arn:aws:sagemaker:ap-southeast-2:452832661640:image/resource-identifier arn:aws:sagemaker:ap-southeast-2:648430277019:image/resource-identifier 648430277019.dkr.ecr.ap-southeast-2.amazonaws.com/sagemaker-distribution-prod:tag
ap-northeast-1 arn:aws:sagemaker:ap-northeast-1:102112518831:image/resource-identifier arn:aws:sagemaker:ap-northeast-1:010972774902:image/resource-identifier 010972774902.dkr.ecr.ap-northeast-1.amazonaws.com/sagemaker-distribution-prod:tag
ca-central-1 arn:aws:sagemaker:ca-central-1:310906938811:image/resource-identifier arn:aws:sagemaker:ca-central-1:481561238223:image/resource-identifier 481561238223.dkr.ecr.ca-central-1.amazonaws.com/sagemaker-distribution-prod:tag
eu-central-1 arn:aws:sagemaker:eu-central-1:936697816551:image/resource-identifier arn:aws:sagemaker:eu-central-1:545423591354:image/resource-identifier 545423591354.dkr.ecr.eu-central-1.amazonaws.com/sagemaker-distribution-prod:tag
eu-west-1 arn:aws:sagemaker:eu-west-1:470317259841:image/resource-identifier arn:aws:sagemaker:eu-west-1:819792524951:image/resource-identifier 819792524951.dkr.ecr.eu-west-1.amazonaws.com/sagemaker-distribution-prod:tag
eu-west-2 arn:aws:sagemaker:eu-west-2:712779665605:image/resource-identifier arn:aws:sagemaker:eu-west-2:021081402939:image/resource-identifier 021081402939.dkr.ecr.eu-west-2.amazonaws.com/sagemaker-distribution-prod:tag
eu-west-3 arn:aws:sagemaker:eu-west-3:615547856133:image/resource-identifier arn:aws:sagemaker:eu-west-3:856416204555:image/resource-identifier 856416204555.dkr.ecr.eu-west-3.amazonaws.com/sagemaker-distribution-prod:tag
eu-north-1 arn:aws:sagemaker:eu-north-1:243637512696:image/resource-identifier arn:aws:sagemaker:eu-north-1:175620155138:image/resource-identifier 175620155138.dkr.ecr.eu-north-1.amazonaws.com/sagemaker-distribution-prod:tag
eu-south-1 arn:aws:sagemaker:eu-south-1:592751261982:image/resource-identifier arn:aws:sagemaker:eu-south-1:810671768855:image/resource-identifier 810671768855.dkr.ecr.eu-south-1.amazonaws.com/sagemaker-distribution-prod:tag
sa-east-1 arn:aws:sagemaker:sa-east-1:782484402741:image/resource-identifier arn:aws:sagemaker:sa-east-1:567556641782:image/resource-identifier 567556641782.dkr.ecr.sa-east-1.amazonaws.com/sagemaker-distribution-prod:tag
ap-northeast-3 arn:aws:sagemaker:ap-northeast-3:792733760839:image/resource-identifier arn:aws:sagemaker:ap-northeast-3:564864627153:image/resource-identifier 564864627153.dkr.ecr.ap-northeast-3.amazonaws.com/sagemaker-distribution-prod:tag
ap-southeast-3 arn:aws:sagemaker:ap-southeast-3:276181064229:image/resource-identifier arn:aws:sagemaker:ap-southeast-3:370607712162:image/resource-identifier 370607712162.dkr.ecr.ap-southeast-3.amazonaws.com/sagemaker-distribution-prod:tag
me-south-1 arn:aws:sagemaker:me-south-1:117516905037:image/resource-identifier arn:aws:sagemaker:me-south-1:523774347010:image/resource-identifier 523774347010.dkr.ecr.me-south-1.amazonaws.com/sagemaker-distribution-prod:tag
me-central-1 arn:aws:sagemaker:me-central-1:103105715889:image/resource-identifier arn:aws:sagemaker:me-central-1:358593528301:image/resource-identifier 358593528301.dkr.ecr.me-central-1.amazonaws.com/sagemaker-distribution-prod:tag

Supported URI tags

The following list shows the tags you can include in your image URI.

  • 1-cpu

  • 1-gpu

  • 0-cpu

  • 0-gpu

The following examples show URIs with various tag formats:

  • 542918446943.dkr.ecr.us-west-2.amazonaws.com/sagemaker-distribution-prod:1-cpu

  • 542918446943.dkr.ecr.us-west-2.amazonaws.com/sagemaker-distribution-prod:0-gpu

Supported images

The following table gives information about the SageMaker images and associated kernels that are available in Amazon SageMaker Studio Classic, as well as the resource identifier and Python version included in the image.

SageMaker images and kernels
SageMaker Image Description Resource Identifier Kernels (and Identifier) Python Version
SageMaker Distribution v0 CPU SageMaker Distribution v0 CPU is a Python 3.8 image that includes popular frameworks for machine learning, data science and visualization on CPU. This includes deep learning frameworks like PyTorch, TensorFlow and Keras; popular Python packages like numpy, scikit-learn and pandas; and IDEs like Jupyter Lab. For more information, see the Amazon SageMaker Distribution repo. sagemaker-distribution-cpu-v0 Python 3 (python3) Python 3.8
SageMaker Distribution v0 GPU SageMaker Distribution v0 GPU is a Python 3.8 image that includes popular frameworks for machine learning, data science and visualization on GPU. This includes deep learning frameworks like PyTorch, TensorFlow and Keras; popular Python packages like numpy, scikit-learn and pandas; and IDEs like Jupyter Lab. For more information, see the Amazon SageMaker Distribution repo. sagemaker-distribution-gpu-v0 Python 3 (python3) Python 3.8
SageMaker Distribution v1 CPU SageMaker Distribution v1 CPU is a Python 3.10 image that includes popular frameworks for machine learning, data science and data analytics on CPU. This includes deep learning frameworks like PyTorch, TensorFlow and Keras; popular Python packages like numpy, scikit-learn and pandas; and IDEs like Jupyter Lab. For more information, see the Amazon SageMaker Distribution repo. sagemaker-distribution-cpu-v1 Python 3 (python3) Python 3.10
SageMaker Distribution v1 GPU SageMaker Distribution v1 GPU is a Python 3.10 image that includes popular frameworks for machine learning, data science and data analytics on GPU. This includes deep learning frameworks like PyTorch, TensorFlow and Keras; popular Python packages like numpy, scikit-learn and pandas; and IDEs like Jupyter Lab. For more information, see the Amazon SageMaker Distribution repo. sagemaker-distribution-gpu-v1 Python 3 (python3) Python 3.10
Base Python 3.0 Official Python 3.10 image from DockerHub with boto3 and Amazon CLI included. sagemaker-base-python-310-v1 Python 3 (python3) Python 3.10
Base Python 2.0 Official Python 3.8 image from DockerHub with boto3 and Amazon CLI included. sagemaker-base-python-38 Python 3 (python3) Python 3.8
Data Science 3.0 Data Science 3.0 is a Python 3.10 conda image based on anaconda version 2022.10 with the most commonly used Python packages and libraries, such as NumPy and SciKit Learn. sagemaker-data-science-310-v1 Python 3 (python3) Python 3.10
Data Science 2.0 Data Science 2.0 is a Python 3.8 conda image based on anaconda version 2021.11 with the most commonly used Python packages and libraries, such as NumPy and SciKit Learn. sagemaker-data-science-38 Python 3 (python3) Python 3.8
Geospatial 1.0 Amazon SageMaker geospatial is a Python image consisting of commonly used geospatial libraries such as GDAL, Fiona, GeoPandas, Shapely, and Rasterio, and allows you to visualize geospatial data within SageMaker. For more information, see Amazon SageMaker geospatial Notebook SDK sagemaker-geospatial-1.0 Python 3 (python3) Python 3.10
PyTorch 2.0.0 Python 3.10 CPU Optimized The Amazon Deep Learning Containers for PyTorch 2.0.0 include containers for training on CPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. pytorch-2.0.0-cpu-py310 Python 3 (python3) Python 3.10
PyTorch 2.0.0 Python 3.10 GPU Optimized The Amazon Deep Learning Containers for PyTorch 2.0.0 with CUDA 11.8 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. pytorch-2.0.0-gpu-py310 Python 3 (python3) Python 3.10
PyTorch 1.13 Python 3.9 CPU Optimized The Amazon Deep Learning Containers for PyTorch 1.13 with CUDA 11.3 include containers for training on CPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. pytorch-1.13-cpu-py39 Python 3 (python3) Python 3.9
PyTorch 1.13 Python 3.9 GPU Optimized The Amazon Deep Learning Containers for PyTorch 1.13 with CUDA 11.7 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. pytorch-1.13-gpu-py39 Python 3 (python3) Python 3.9
PyTorch 1.12 Python 3.8 CPU Optimized The Amazon Deep Learning Containers for PyTorch 1.12 with CUDA 11.3 include containers for training on CPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers for PyTorch 1.12.0. pytorch-1.12-cpu-py38 Python 3 (python3) Python 3.8
PyTorch 1.12 Python 3.8 GPU Optimized The Amazon Deep Learning Containers for PyTorch 1.12 with CUDA 11.3 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers for PyTorch 1.12.0. pytorch-1.12-gpu-py38 Python 3 (python3) Python 3.8
PyTorch 1.10 Python 3.8 CPU Optimized The Amazon Deep Learning Containers for PyTorch 1.10 include containers for training on CPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers for PyTorch 1.10.2 on SageMaker. pytorch-1.10-cpu-py38 Python 3 (python3) Python 3.8
PyTorch 1.10 Python 3.8 GPU Optimized The Amazon Deep Learning Containers for PyTorch 1.10 with CUDA 11.3 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers for PyTorch 1.10.2 on SageMaker. pytorch-1.10-gpu-py38 Python 3 (python3) Python 3.8
SparkAnalytics 2.0 Anaconda Individual Edition with PySpark and Spark kernels. For more information, see sparkmagic. sagemaker-sparkanalytics-310-v1
  • SparkMagic Spark (conda-env-sm_sparkmagic-sparkkernel)

  • SparkMagic PySpark (conda-env-sm_sparkmagic-pysparkkernel)

  • Glue Spark (conda-env-sm_glue_is-glue_spark)

  • Glue Python [PySpark and Ray] (conda-env-sm_glue_is-glue_pyspark)

Python 3.10
SparkAnalytics 1.0 Anaconda Individual Edition with PySpark and Spark kernels. For more information, see sparkmagic. sagemaker-sparkanalytics-v1
  • SparkMagic Spark (conda-env-sm_sparkmagic-sparkkernel)

  • SparkMagic PySpark (conda-env-sm_sparkmagic-pysparkkernel)

  • Glue Spark (conda-env-sm_glue_is-glue_spark)

  • Glue Python [PySpark and Ray] (conda-env-sm_glue_is-glue_pyspark)

Python 3.8
TensorFlow 2.12.0 Python 3.10 CPU Optimized The Amazon Deep Learning Containers for TensorFlow 2.12.0 with CUDA 11.2 include containers for training on CPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. tensorflow-2.12.0-cpu-py310-ubuntu20.04-sagemaker-v1.0 Python 3 (python3) Python 3.10
TensorFlow 2.12.0 Python 3.10 GPU Optimized The Amazon Deep Learning Containers for TensorFlow 2.12.0 with CUDA 11.8 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. tensorflow-2.12.0-gpu-py310-cu118-ubuntu20.04-sagemaker-v1 Python 3 (python3) Python 3.10
TensorFlow 2.11.0 Python 3.9 CPU Optimized The Amazon Deep Learning Containers for TensorFlow 2.11.0 with CUDA 11.2 include containers for training on CPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. tensorflow-2.11.0-cpu-py39-ubuntu20.04-sagemaker-v1.1 Python 3 (python3) Python 3.9
TensorFlow 2.11.0 Python 3.9 GPU Optimized The Amazon Deep Learning Containers for TensorFlow 2.11.0 with CUDA 11.2 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. tensorflow-2.11.0-gpu-py39-cu112-ubuntu20.04-sagemaker-v1.1 Python 3 (python3) Python 3.9
TensorFlow 2.10 Python 3.9 CPU Optimized The Amazon Deep Learning Containers for TensorFlow 2.10 with CUDA 11.2 include containers for training on CPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. tensorflow-2.10.1-cpu-py39-ubuntu20.04-sagemaker-v1.2 Python 3 (python3) Python 3.9
TensorFlow 2.10 Python 3.9 GPU Optimized The Amazon Deep Learning Containers for TensorFlow 2.10 with CUDA 11.2 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Release Notes for Deep Learning Containers. tensorflow-2.10.1-gpu-py39-ubuntu20.04-sagemaker-v1.2 Python 3 (python3) Python 3.9
TensorFlow 2.6 Python 3.8 CPU Optimized The Amazon Deep Learning Containers for TensorFlow 2.6 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers for TensorFlow 2.6. tensorflow-2.6-cpu-py38-ubuntu20.04-v1 Python 3 (python3) Python 3.8
TensorFlow 2.6 Python 3.8 GPU Optimized The Amazon Deep Learning Containers for TensorFlow 2.6 with CUDA 11.2 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers for TensorFlow 2.6. tensorflow-2.6-gpu-py38-cu112-ubuntu20.04-v1 Python 3 (python3) Python 3.8

Images slated for deprecation

SageMaker ends support for images the day after any of the packages in the image reach end-of life by their publisher.

The following SageMaker images are slated for deprecation. These images are based on Python 3.7, which reached end-of-life on June 27, 2023. Starting on October 30, 2023, SageMaker will discontinue support for these images and they will not be selectable from the Studio Classic UI. To avoid non-compliance issues, if you're using any of these images, we recommend that you move to an image with a later version.

SageMaker images slated for deprecation
SageMaker Image Deprecation date Description Resource Identifier Kernels Python Version
Data Science October 30, 2023 Data Science is a Python 3.7 conda image with the most commonly used Python packages and libraries, such as NumPy and SciKit Learn. datascience-1.0 Python 3 Python 3.7
SageMaker JumpStart Data Science 1.0 October 30, 2023 SageMaker JumpStart Data Science 1.0 is a SageMaker JumpStart image that includes commonly used packages and libraries. sagemaker-jumpstart-data-science-1.0 Python 3 Python 3.7
SageMaker JumpStart MXNet 1.0 October 30, 2023 SageMaker JumpStart MXNet 1.0 is a SageMaker JumpStart image that includes MXNet. sagemaker-jumpstart-mxnet-1.0 Python 3 Python 3.7
SageMaker JumpStart PyTorch 1.0 October 30, 2023 SageMaker JumpStart PyTorch 1.0 is a SageMaker JumpStart image that includes PyTorch. sagemaker-jumpstart-pytorch-1.0 Python 3 Python 3.7
SageMaker JumpStart TensorFlow 1.0 October 30, 2023 SageMaker JumpStart TensorFlow 1.0 is a SageMaker JumpStart image that includes TensorFlow. sagemaker-jumpstart-tensorflow-1.0 Python 3 Python 3.7
SparkMagic October 30, 2023 Anaconda Individual Edition with PySpark and Spark kernels. For more information, see sparkmagic. sagemaker-sparkmagic
  • PySpark

  • Spark

Python 3.7
TensorFlow 2.3 Python 3.7 CPU Optimized October 30, 2023 The Amazon Deep Learning Containers for TensorFlow 2.3 include containers for training on CPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers with TensorFlow 2.3.0. tensorflow-2.3-cpu-py37-ubuntu18.04-v1 Python 3 Python 3.7
TensorFlow 2.3 Python 3.7 GPU Optimized October 30, 2023 The Amazon Deep Learning Containers for TensorFlow 2.3 with CUDA 11.0 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers for TensorFlow 2.3.1 with CUDA 11.0. tensorflow-2.3-gpu-py37-cu110-ubuntu18.04-v3 Python 3 Python 3.7
TensorFlow 1.15 Python 3.7 CPU Optimized October 30, 2023 The Amazon Deep Learning Containers for TensorFlow 1.15 include containers for training on CPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers v7.0 for TensorFlow. tensorflow-1.15-cpu-py37-ubuntu18.04-v7 Python 3 Python 3.7
TensorFlow 1.15 Python 3.7 GPU Optimized October 30, 2023 The Amazon Deep Learning Containers for TensorFlow 1.15 with CUDA 11.0 include containers for training on GPU, optimized for performance and scale on Amazon. For more information, see Amazon Deep Learning Containers v7.0 for TensorFlow. tensorflow-1.15-gpu-py37-cu110-ubuntu18.04-v8 Python 3 Python 3.7