PyTorch Deep Learning Containers TensorFlow 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.18	Training	EC2 ECS EKS	X86	AWS Deep Learning Containers for TensorFlow 2.18 (Training on EC2, ECS, and EKS): January 29, 2025
2.18	Training	SageMaker	X86	AWS Deep Learning Containers for TensorFlow 2.18 (Training on SageMaker AI): January 29, 2025
2.18	Inference	EC2 ECS EKS	X86	AWS Deep Learning Containers for TensorFlow 2.18 (Inference on EC2, ECS, and EKS): January 29, 2025
2.18	Inference	SageMaker	X86	AWS Deep Learning Containers for TensorFlow 2.18 (Inference on SageMaker AI): January 29, 2025
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

Warning Javascript is disabled or is unavailable in your browser.

To use the Amazon Web Services Documentation, Javascript must be enabled. Please refer to your browser's Help pages for instructions.

Document Conventions

Troubleshooting AWS Deep Learning Containers on EKS

Release Notifications