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.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): 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): October 3, 2024
2.4	Inference	EC2 ECS EKS	Arm64	AWS Deep Learning Containers for Graviton PyTorch 2.4 (Inference on EC2, ECS, and EKS): July 11, 2024
2.4	Inference	SageMaker	Arm64	AWS Deep Learning Containers for Graviton PyTorch 2.4 (SageMaker): 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): 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): July 11, 2024
2.3	Inference	EC2 ECS EKS	Arm64	AWS Deep Learning Containers for Graviton PyTorch 2.3 (EC2, ECS, and EKS): July 11, 2024
2.3	Inference	SageMaker	Arm64	AWS Deep Learning Containers for Graviton PyTorch 2.3 (SageMaker): 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): 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): March 20, 2024
2.2	Inference	EC2 ECS EKS	Arm64	AWS Deep Learning Containers for PyTorch 2.2 (Inference on EC2, ECS, and EKS): March 20, 2024
2.2	Inference	SageMaker	Arm64	AWS Deep Learning Containers for Graviton PyTorch 2.2 (EC2, ECS, and EKS): April 16, 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): 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): July 19, 2024
2.16	Inference	EC2 ECS EKS	Arm64	AWS Deep Learning Containers for Graviton TensorFlow 2.16 (EC2, ECS, and EKS): July 19, 2024
2.16	Inference	SageMaker	Arm64	AWS Deep Learning Containers for Graviton TensorFlow 2.16 (SageMaker): 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