Amazon SageMaker AI images available for use with Studio Classic - Amazon SageMaker AI
Image ARN formatSupported URI tagsSupported imagesImages slated for deprecationDeprecated images

Amazon SageMaker AI images available for use with Studio Classic

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 AI images and associated kernels that are available in Amazon SageMaker Studio Classic. This page also gives information about the format needed to create the ARN for each image. SageMaker AI 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. The resource identifier is found in the SageMaker AI 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 AI images and associated kernels that are available in Amazon SageMaker Studio Classic. It also gives information about the resource identifier and Python version included in the image.

SageMaker AI images and kernels

SageMaker AI Image Description Resource Identifier Kernels (and Identifier) Python Version
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 AWS CLI included. sagemaker-base-python-310-v1 Python 3 (python3) Python 3.10
Data Science 4.0 Data Science 4.0 is a Python 3.11 conda image based on Ubuntu version 22.04. It includes the most commonly used Python packages and libraries, such as NumPy and SciKit Learn. sagemaker-data-science-311-v1 Python 3 (python3) Python 3.11
Data Science 3.0 Data Science 3.0 is a Python 3.10 conda image based on Ubuntu version 22.04. It includes 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
Geospatial 1.0 Amazon SageMaker geospatial is a Python image consisting of commonly used geospatial libraries such as GDAL, Fiona, GeoPandas, Shapley, and Rasterio. It allows you to visualize geospatial data within SageMaker AI. For more information, see Amazon SageMaker geospatial Notebook SDK sagemaker-geospatial-1.0 Python 3 (python3) Python 3.10
SparkAnalytics 3.0 The SparkAnalytics 3.0 image provides Spark and PySpark kernel options on Amazon SageMaker Studio Classic, including SparkMagic Spark, SparkMagic PySpark, Glue Spark, and Glue PySpark, enabling flexible distributed data processing. sagemaker-sparkanalytics-311-v1

  • SparkMagic Spark (sparkkernel)

  • SparkMagic PySpark (pysparkkernel)

  • Glue Spark (glue_spark)

  • Glue PySpark (glue_pyspark)

 Python 3.11
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
PyTorch 2.4.0 Python 3.11 CPU Optimized The AWS Deep Learning Containers for PyTorch 2.4.0 with CUDA 12.4 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.4.0-cpu-py311 Python 3 (python3) Python 3.11
PyTorch 2.4.0 Python 3.11 GPU Optimized The AWS Deep Learning Containers for PyTorch 2.4.0 with CUDA 12.4 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.4.0-gpu-py311 Python 3 (python3) Python 3.11
PyTorch 2.3.0 Python 3.11 CPU Optimized The AWS Deep Learning Containers for PyTorch 2.3.0 with CUDA 12.1 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.3.0-cpu-py311 Python 3 (python3) Python 3.11
PyTorch 2.3.0 Python 3.11 GPU Optimized The AWS Deep Learning Containers for PyTorch 2.3.0 with CUDA 12.1 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.3.0-gpu-py311 Python 3 (python3) Python 3.11
PyTorch 2.2.0 Python 3.10 CPU Optimized The AWS Deep Learning Containers for PyTorch 2.2 with CUDA 12.1 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.2.0-cpu-py310 Python 3 (python3) Python 3.10
PyTorch 2.2.0 Python 3.10 GPU Optimized The AWS Deep Learning Containers for PyTorch 2.2 with CUDA 12.1 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.2.0-gpu-py310 Python 3 (python3) Python 3.10
PyTorch 2.1.0 Python 3.10 CPU Optimized The AWS Deep Learning Containers for PyTorch 2.1 with CUDA 12.1 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.1.0-cpu-py310 Python 3 (python3) Python 3.10
PyTorch 2.1.0 Python 3.10 GPU Optimized The AWS Deep Learning Containers for PyTorch 2.1 with CUDA 12.1 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.1.0-gpu-py310 Python 3 (python3) Python 3.10
PyTorch 1.13 HuggingFace Python 3.10 Neuron Optimized PyTorch 1.13 image with HuggingFace and Neuron packages installed for training on Trainium instances optimized for performance and scale on AWS. pytorch-1.13-hf-neuron-py310 Python 3 (python3) Python 3.10
PyTorch 1.13 Python 3.10 Neuron Optimized PyTorch 1.13 image with Neuron packages installed for training on Trainium instances optimized for performance and scale on AWS. pytorch-1.13-neuron-py310 Python 3 (python3) Python 3.10
TensorFlow 2.14.0 Python 3.10 CPU Optimized The AWS Deep Learning Containers for TensorFlow 2.14 with CUDA 11.8 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. tensorflow-2.14.1-cpu-py310-ubuntu20.04-sagemaker-v1.0 Python 3 (python3) Python 3.10
TensorFlow 2.14.0 Python 3.10 GPU Optimized The AWS Deep Learning Containers for TensorFlow 2.14 with CUDA 11.8 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. tensorflow-2.14.1-gpu-py310-cu118-ubuntu20.04-sagemaker-v1.0 Python 3 (python3) Python 3.10

Images slated for deprecation

SageMaker AI ends support for images the day after any of the packages in the image reach end-of life by their publisher. The following SageMaker AI images are slated for deprecation.

Images based on Python 3.8 reached end-of-life on October 31st, 2024. Starting on November 1, 2024, SageMaker AI 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 AI images slated for deprecation

SageMaker AI Image Deprecation date Description Resource Identifier Kernels Python Version
SageMaker Distribution v0.12 CPU November 1, 2024 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 AI Distribution repo. sagemaker-distribution-cpu-v0 Python 3 (python3) Python 3.8
SageMaker Distribution v0.12 GPU November 1, 2024 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 AI Distribution repo. sagemaker-distribution-gpu-v0 Python 3 (python3) Python 3.8
Base Python 2.0 November 1, 2024 Official Python 3.8 image from DockerHub with boto3 and AWS CLI included. sagemaker-base-python-38 Python 3 (python3) Python 3.8
Data Science 2.0 November 1, 2024 Data Science 2.0 is a Python 3.8 conda image based on Ubuntu version 22.04. It includes the most commonly used Python packages and libraries, such as NumPy and SciKit Learn. sagemaker-data-science-38 Python 3 (python3) Python 3.8
PyTorch 1.13 Python 3.9 CPU Optimized November 1, 2024 The AWS Deep Learning Containers for PyTorch 1.13 with CUDA 11.3 include containers for training on CPU, optimized for performance and scale on AWS. 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 November 1, 2024 The AWS Deep Learning Containers for PyTorch 1.13 with CUDA 11.7 include containers for training on GPU, optimized for performance and scale on AWS. 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 November 1, 2024 The AWS Deep Learning Containers for PyTorch 1.12 with CUDA 11.3 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see AWS 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 November 1, 2024 The AWS Deep Learning Containers for PyTorch 1.12 with CUDA 11.3 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see AWS 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 November 1, 2024 The AWS Deep Learning Containers for PyTorch 1.10 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see AWS Deep Learning Containers for PyTorch 1.10.2 on SageMaker AI. pytorch-1.10-cpu-py38 Python 3 (python3) Python 3.8
PyTorch 1.10 Python 3.8 GPU Optimized November 1, 2024 The AWS Deep Learning Containers for PyTorch 1.10 with CUDA 11.3 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see AWS Deep Learning Containers for PyTorch 1.10.2 on SageMaker AI. pytorch-1.10-gpu-py38 Python 3 (python3) Python 3.8
SparkAnalytics 1.0 November 1, 2024 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.13.0 Python 3.10 CPU Optimized November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.13 with CUDA 11.8 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers.. tensorflow-2.13.0-cpu-py310-ubuntu20.04-sagemaker-v1.0 Python 3 (python3) Python 3.10
TensorFlow 2.13.0 Python 3.10 GPU Optimized November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.13 with CUDA 11.8 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. tensorflow-2.13.0-gpu-py310-cu118-ubuntu20.04-sagemaker-v1.0 Python 3 (python3) Python 3.10
TensorFlow 2.6 Python 3.8 CPU Optimized November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.6 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see AWS 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 November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.6 with CUDA 11.2 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see AWS Deep Learning Containers for TensorFlow 2.6. tensorflow-2.6-gpu-py38-cu112-ubuntu20.04-v1 Python 3 (python3) Python 3.8
PyTorch 2.0.1 Python 3.10 CPU Optimized November 1, 2024 The AWS Deep Learning Containers for PyTorch 2.0.1 with CUDA 12.1 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.0.1-cpu-py310 Python 3 (python3) Python 3.10
PyTorch 2.0.1 Python 3.10 GPU Optimized November 1, 2024 The AWS Deep Learning Containers for PyTorch 2.0.1 with CUDA 12.1 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.0.1-gpu-py310 Python 3 (python3) Python 3.10
PyTorch 2.0.0 Python 3.10 CPU Optimized November 1, 2024 The AWS Deep Learning Containers for PyTorch 2.0.0 include containers for training on CPU, optimized for performance and scale on AWS. 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 November 1, 2024 The AWS Deep Learning Containers for PyTorch 2.0.0 with CUDA 11.8 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see Release Notes for Deep Learning Containers. pytorch-2.0.0-gpu-py310 Python 3 (python3) Python 3.10
TensorFlow 2.12.0 Python 3.10 CPU Optimized November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.12.0 with CUDA 11.2 include containers for training on CPU, optimized for performance and scale on AWS. 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 November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.12.0 with CUDA 11.8 include containers for training on GPU, optimized for performance and scale on AWS. 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 November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.11.0 with CUDA 11.2 include containers for training on CPU, optimized for performance and scale on AWS. 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 November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.11.0 with CUDA 11.2 include containers for training on GPU, optimized for performance and scale on AWS. 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 November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.10 with CUDA 11.2 include containers for training on CPU, optimized for performance and scale on AWS. 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 November 1, 2024 The AWS Deep Learning Containers for TensorFlow 2.10 with CUDA 11.2 include containers for training on GPU, optimized for performance and scale on AWS. 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

Deprecated images

SageMaker AI has ended support for the following images. Deprecaiton occurs the day after any of the packages in the image reach end-of life by their publisher.

SageMaker AI images slated for deprecation

SageMaker AI 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 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 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 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 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 AWS Deep Learning Containers for TensorFlow 2.3 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see AWS 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 AWS Deep Learning Containers for TensorFlow 2.3 with CUDA 11.0 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see AWS 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 AWS Deep Learning Containers for TensorFlow 1.15 include containers for training on CPU, optimized for performance and scale on AWS. For more information, see AWS 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 AWS Deep Learning Containers for TensorFlow 1.15 with CUDA 11.0 include containers for training on GPU, optimized for performance and scale on AWS. For more information, see AWS Deep Learning Containers v7.0 for TensorFlow. tensorflow-1.15-gpu-py37-cu110-ubuntu18.04-v8 Python 3 Python 3.7