Amazon Deep Learning Containers for TensorFlow 2.18 ARM64 Inference on SageMaker - Amazon Deep Learning Containers
Release NotesSecurity Advisory Python Support CPU Instance Type Support Amazon Regions support Build and Test
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).

Amazon Deep Learning Containers for TensorFlow 2.18 ARM64 Inference on SageMaker

Amazon Deep Learning Containers (DLCs) for Amazon SageMaker, are now available for ARM64 platforms, including the Graviton instance types with support for TensorFlow Serving 2.18. You can launch the new versions of the Deep Learning Containers on SageMaker.

This release includes a container image for inference on CPU optimized for performance and scale on Amazon. This Docker image was tested on Amazon SageMaker. It provides an optimized user experience for running deep learning workloads on SageMaker. All software components in this image are scanned for security vulnerabilities and updated or patched in accordance with Amazon Security best practices.

A list of available containers can be found in our documentation. Get started quickly with the Amazon Deep Learning Containers using the getting-started guides and beginner to advanced level tutorials in our developer guide. You can also subscribe to our discussion forum to get launch announcements and post your questions.

Release Notes

  • Support for TensorFlow Serving 2.18 on Amazon SageMaker. For details about this release, checkout our GitHub v1.0-tf-arm64-sagemaker-2.18.0-inf-cpu-py310

  • Starting with TensorFlow 2.18, we are changing the name of Graviton DLCs to ARM64 DLCs in order to generalize the usage. For example, the ECR repository name is now "tensorflow-inference-arm64" instead of "tensorflow-inference-graviton". Graviton DLCs and ARM64 DLCs are functionally equivalent.

For latest updates, please refer to the aws/deep-learning-containers GitHub repo.

Security Advisory

Python Support

Python 3.10 is supported in the containers for the installed deep learning frameworks.

CPU Instance Type Support

The container supports Graviton CPU instance types.

Amazon Regions support

The containers are available in the following regions:

Region

Code

US East (Ohio)

us-east-2

US East (N. Virginia)

us-east-1

US West (Oregon)

us-west-2

US West (N. California)

us-west-1

AF South (Cape Town)

af-south-1

Asia Pacific (Hong Kong)

ap-east-1

Asia Pacific (Hyderabad)

ap-south-2

Asia Pacific (Mumbai)

ap-south-1

Asia Pacific (Osaka)

ap-northeast-3

Asia Pacific (Seoul)

ap-northeast-2

Asia Pacific (Tokyo)

ap-northeast-1

Asia Pacific (Melbourne)

ap-southeast-4

Asia Pacific (Jakarta)

ap-southeast-3

Asia Pacific (Sydney)

ap-southeast-2

Asia Pacific (Singapore)

ap-southeast-1

Asia Pacific (Malaysia)

ap-southeast-5

Central (Canada)

ca-central-1

Canada (Calgary)

ca-west-1

EU (Zurich)

eu-central-2

EU (Frankfurt)

eu-central-1

EU (Ireland)

eu-west-1

EU (London)

eu-west-2

EU( Paris)

eu-west-3

EU (Spain)

eu-south-2

EU (Milan)

eu-south-1

EU (Stockholm)

eu-north-1

Israel (Tel Aviv)

il-central-1

Middle East (Bahrain)

me-south-1

Middle East (UAE)

me-central-1

SA (Sau Paulo)

sa-east-1

China (Beijing)

cn-north-1

China (Ningxia)

cn-northwest-1

Build and Test

  • Built on: c6g.2xlarge

  • DLC Images tested on: c8g.4xlarge, t4g.2xlarge, r8g.2xlarge, m7g.4xlarge