PyTorch Release Notes¶
Release notes for AWS Deep Learning Containers with PyTorch.
PyTorch v2 (Amazon Linux 2023)¶
| Framework | Version | Accelerator | Platform | Link |
|---|---|---|---|---|
| PyTorch | 2.11 | GPU | SageMaker | Release Notes |
| PyTorch | 2.11 | CPU | SageMaker | Release Notes |
| PyTorch | 2.11 | GPU | EC2, ECS, EKS | Release Notes |
| PyTorch | 2.11 | CPU | EC2, ECS, EKS | Release Notes |
PyTorch 2.10 (Ubuntu)¶
| Framework | Version | Accelerator | Platform | Link |
|---|---|---|---|---|
| PyTorch Training | 2.10 | GPU | SageMaker | Release Notes |
| PyTorch Training | 2.10 | CPU | SageMaker | Release Notes |
| PyTorch Training | 2.10 | GPU | EC2, ECS, EKS | Release Notes |
| PyTorch Training | 2.10 | CPU | EC2, ECS, EKS | Release Notes |
PyTorch 2.9 (Ubuntu)¶
| 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 (Ubuntu)¶
| 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.6 (Ubuntu)¶
| 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.7 (Ubuntu)¶
| 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 (Ubuntu)¶
| 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 |