Release Notes for Deep Learning Containers - AWS Deep Learning Containers

Release Notes for Deep Learning Containers

Check the latest release notes for AWS Deep Learning Containers built for specific machine learning frameworks, infrastructures, and AWS services.

Note

Starting with MXNet 1.9, PyTorch 1.10, and Tensorflow 2.7, CPU and GPU Deep Learning Containers are released as either E3 or SageMaker images. E3 images are supported on Amazon EC2, Amazon ECS, and Amazon EKS, whereas SageMaker images are supported on Amazon SageMaker and contain additional packages to support SageMaker workloads.

Single-framework Deep Learning Containers

TensorFlow

PyTorch

AWS MX powered by Apache MXNet

Graviton Deep Learning Containers

TensorFlow

PyTorch

Habana Deep Learning Containers

TensorFlow

PyTorch