PyTorch Release Notes¶
Release notes for AWS Deep Learning Containers with PyTorch.
PyTorch 2.9¶
| Framework | Version | Accelerator | Platform | Link |
|---|---|---|---|---|
| PyTorch Training | 2.9 | GPU | SageMaker | Release Notes |
| PyTorch Training | 2.9 | CPU | SageMaker | Release Notes |
| PyTorch Training | 2.9 | GPU | EC2, ECS, EKS | Release Notes |
| PyTorch Training | 2.9 | CPU | EC2, ECS, EKS | Release Notes |
PyTorch 2.8¶
| Framework | Version | Accelerator | Platform | Link |
|---|---|---|---|---|
| PyTorch Training | 2.8 | GPU | SageMaker | Release Notes |
| PyTorch Training | 2.8 | CPU | SageMaker | Release Notes |
| PyTorch Training | 2.8 | GPU | EC2, ECS, EKS | Release Notes |
| PyTorch Training | 2.8 | CPU | EC2, ECS, EKS | Release Notes |
PyTorch 2.7¶
| Framework | Version | Accelerator | Platform | Link |
|---|---|---|---|---|
| PyTorch Training | 2.7 | GPU | SageMaker | Release Notes |
| PyTorch Training | 2.7 | CPU | SageMaker | Release Notes |
| PyTorch Training | 2.7 | GPU | EC2, ECS, EKS | Release Notes |
| PyTorch Training | 2.7 | CPU | EC2, ECS, EKS | Release Notes |
| PyTorch Training ARM64 | 2.7 | GPU | EC2, ECS, EKS | Release Notes |
PyTorch 2.6¶
| Framework | Version | Accelerator | Platform | Link |
|---|---|---|---|---|
| PyTorch Inference | 2.6 | GPU | SageMaker | Release Notes |
| PyTorch Inference | 2.6 | CPU | SageMaker | Release Notes |
| PyTorch Inference | 2.6 | GPU | EC2, ECS, EKS | Release Notes |
| PyTorch Inference | 2.6 | CPU | EC2, ECS, EKS | Release Notes |
| PyTorch Inference ARM64 | 2.6 | CPU | SageMaker | Release Notes |
| PyTorch Inference ARM64 | 2.6 | GPU | EC2, ECS, EKS | Release Notes |
| PyTorch Inference ARM64 | 2.6 | CPU | EC2, ECS, EKS | Release Notes |
Deprecated Images
The following images are no longer supported and do not receive security patches or bug fixes.
PyTorch 2.6¶
| Framework | Version | Accelerator | Platform | Link |
|---|---|---|---|---|
| PyTorch Training | 2.6 | GPU | SageMaker | Release Notes |
| PyTorch Training | 2.6 | CPU | SageMaker | Release Notes |
| PyTorch Training | 2.6 | GPU | EC2, ECS, EKS | Release Notes |
| PyTorch Training | 2.6 | CPU | EC2, ECS, EKS | Release Notes |