Release Notes - AWS Deep Learning Containers
PyTorch Deep Learning ContainersTensorFlow Deep Learning Containers

Release Notes

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

Tip

For the full list of available Deep Learning Containers and information on pulling them, see Available Deep Learning Containers Images.

PyTorch Deep Learning Containers

Version Type Service Architecture Release Note
2.5 Training EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.5 (Training on EC2, ECS, and EKS): November 25, 2024
2.5 Training SageMaker X86

AWS Deep Learning Containers for PyTorch 2.5 (Training on SageMaker AI): November 25, 2024
2.5 Inference EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.5 (Inference on EC2, ECS, and EKS): November 13, 2024
2.5 Inference SageMaker X86

AWS Deep Learning Containers for PyTorch 2.5 (Inference on SageMaker AI): November 13, 2024
2.5 Inference EC2 ECS EKS Arm64

AWS Deep Learning Containers for PyTorch 2.5 ARM64 (Inference on EC2, ECS, and EKS): December 11, 2024
2.5 Inference SageMaker Arm64

AWS Deep Learning Containers for PyTorch 2.5 ARM64 (Inference on SageMaker AI): December 11, 2024
2.4 Training EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.4 (Training on EC2, ECS, and EKS): September 26, 2024
2.4 Training SageMaker X86

AWS Deep Learning Containers for PyTorch 2.4 (Training on SageMaker AI): September 26, 2024
2.4 Inference EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.4 (Inference on EC2, ECS, and EKS): September 26, 2024
2.4 Inference SageMaker X86

AWS Deep Learning Containers for PyTorch 2.4 (Inference on SageMaker AI): October 3, 2024
2.4 Inference EC2 ECS EKS Arm64

AWS Deep Learning Containers for PyTorch 2.4 Graviton (Inference on EC2, ECS, and EKS): July 11, 2024
2.4 Inference SageMaker Arm64

AWS Deep Learning Containers for PyTorch 2.4 Graviton (Inference on SageMaker AI): July 11, 2024
2.3 Training EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.3 (Training on EC2, ECS, and EKS): May 30, 2024
2.3 Training SageMaker X86

AWS Deep Learning Containers for PyTorch 2.3 (Training on SageMaker AI): May 30, 2024
2.3 Inference EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.3 (Inference on EC2, ECS, and EKS): July 11, 2024
2.3 Inference SageMaker X86

AWS Deep Learning Containers for PyTorch 2.3 (Inference on SageMaker AI): July 11, 2024
2.3 Inference EC2 ECS EKS Arm64

AWS Deep Learning Containers for PyTorch 2.3 Graviton (Inference on EC2, ECS, and EKS): July 11, 2024
2.3 Inference SageMaker Arm64

AWS Deep Learning Containers for PyTorch 2.3 Graviton (Inference on SageMaker AI): July 11, 2024
2.2 Training EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.2 (Training on EC2, ECS, and EKS): March 20, 2024
2.2 Training SageMaker X86

AWS Deep Learning Containers for PyTorch 2.2 (Training on SageMaker AI): March 20, 2024
2.2 Inference EC2 ECS EKS X86

AWS Deep Learning Containers for PyTorch 2.2 (Inference on EC2, ECS, and EKS): March 20, 2024
2.2 Inference SageMaker X86

AWS Deep Learning Containers for PyTorch 2.2 (Inference on SageMaker AI): March 20, 2024
2.2 Inference EC2 ECS EKS Arm64

AWS Deep Learning Containers for PyTorch 2.2 Graviton (Inference on EC2, ECS, and EKS): April 25, 2024
2.2 Inference SageMaker Arm64

AWS Deep Learning Containers for PyTorch 2.2 Graviton (Inference on SageMaker AI): April 25, 2024

TensorFlow Deep Learning Containers

Version Type Service Architecture Release Note
2.16 Training EC2 ECS EKS X86

AWS Deep Learning Containers for TensorFlow 2.16 (Training on EC2, ECS, and EKS): August 20, 2024
2.16 Training SageMaker X86

AWS Deep Learning Containers for TensorFlow 2.16 (Training on SageMaker AI): August 20, 2024
2.16 Inference EC2 ECS EKS X86

AWS Deep Learning Containers for TensorFlow 2.16 (Inference on EC2, ECS, and EKS): July 19, 2024
2.16 Inference SageMaker X86

AWS Deep Learning Containers for TensorFlow 2.16 (Inference on SageMaker AI): July 19, 2024
2.16 Inference EC2 ECS EKS Arm64

AWS Deep Learning Containers for TensorFlow 2.16 Graviton (Inference on EC2, ECS, and EKS): July 19, 2024
2.16 Inference SageMaker Arm64

AWS Deep Learning Containers for TensorFlow 2.16 Graviton (Inference on SageMaker AI): July 19, 2024