Skip to content

Commit 3bb6ae8

Browse files
Yadan-WeiYadan Wei
Yadan-Wei
and
Yadan Wei
authored
add PT2.6 Training to Available Images (#4630)
Co-authored-by: Yadan Wei <[email protected]>
1 parent ea48b42 commit 3bb6ae8

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

‎available_images.md

+8-4
Original file line numberDiff line numberDiff line change
@@ -101,8 +101,8 @@ EC2 Framework Containers (EC2, ECS, and EKS support only)
101101

102102
| Framework |Job Type |Horovod Options|CPU/GPU |Python Version Options | Example URL |
103103
|-------------------|-----------|---------------|-----------|-----------------------|-----------------------------------------------------------------------------------------------------------|
104-
| PyTorch 2.5.1 |training |No |CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.5.1-cpu-py311-ubuntu22.04-ec2 |
105-
| PyTorch 2.5.1 |training |No |GPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.5.1-gpu-py311-cu124-ubuntu22.04-ec2 |
104+
| PyTorch 2.6.0 |training |No |CPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.6.0-cpu-py312-ubuntu22.04-ec2 |
105+
| PyTorch 2.6.0 |training |No |GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.6.0-gpu-py312-cu126-ubuntu22.04-ec2 |
106106
| PyTorch 2.6.0 |inference |No |CPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.6.0-cpu-py312-ubuntu22.04-ec2 |
107107
| PyTorch 2.6.0 |inference |No |GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.6.0-gpu-py312-cu124-ubuntu22.04-ec2 |
108108
| TensorFlow 2.18.0 |training |No |CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.18.0-cpu-py310-ubuntu22.04-ec2 |
@@ -116,8 +116,8 @@ SageMaker Framework Containers (SM support only)
116116

117117
| Framework | Job Type | Horovod Options | CPU/GPU | Python Version Options | Example URL |
118118
|-------------------|-----------|-----------------|-----------|------------------------|----------------------------------------------------------------------------------------------------------------|
119-
| PyTorch 2.5.1 | training | No | CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.5.1-cpu-py311-ubuntu22.04-sagemaker |
120-
| PyTorch 2.5.1 | training | No | GPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.5.1-gpu-py311-cu124-ubuntu22.04-sagemaker |
119+
| PyTorch 2.6.0 | training | No | CPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.6.0-cpu-py312-ubuntu22.04-sagemaker |
120+
| PyTorch 2.6.0 | training | No | GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.6.0-gpu-py312-cu126-ubuntu22.04-sagemaker |
121121
| PyTorch 2.6.0 | inference | No | CPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.6.0-cpu-py312-ubuntu22.04-sagemaker |
122122
| PyTorch 2.6.0 | inference | No | GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.6.0-gpu-py312-cu124-ubuntu22.04-sagemaker |
123123
| TensorFlow 2.18.0 | training | No | CPU | 3.10 (py310) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.18.0-cpu-py310-ubuntu22.04-sagemaker |
@@ -324,6 +324,8 @@ Prior EC2 Framework Container Versions
324324
==============
325325
| Framework |Job Type |Horovod Options |CPU/GPU |Python Version Options |Example URL |
326326
|---------------------------------------------|------------------------|---------------------------------|------------|---------------------------------------|----------------------------------------------------------------------------------------------------|
327+
| PyTorch 2.5.1 |training |No |CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.5.1-cpu-py311-ubuntu22.04-ec2 |
328+
| PyTorch 2.5.1 |training |No |GPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.5.1-gpu-py311-cu124-ubuntu22.04-ec2 |
327329
| PyTorch 2.5.1 |inference |No |CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.5.1-cpu-py311-ubuntu22.04-ec2 |
328330
| PyTorch 2.5.1 |inference |No |GPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.5.1-gpu-py311-cu124-ubuntu22.04-ec2 |
329331
| PyTorch 2.4.0 |training |No |CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.4.0-cpu-py311-ubuntu22.04-ec2 |
@@ -339,6 +341,8 @@ Prior SageMaker Framework Container Versions
339341
==============
340342
| Framework |Job Type |Horovod Options |CPU/GPU |Python Version Options |Example URL |
341343
|---------------------------------------------|------------------------|---------------------------------|------------|---------------------------------------|----------------------------------------------------------------------------------------------------|
344+
| PyTorch 2.5.1 | training | No | CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.5.1-cpu-py311-ubuntu22.04-sagemaker |
345+
| PyTorch 2.5.1 | training | No | GPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.5.1-gpu-py311-cu124-ubuntu22.04-sagemaker |
342346
| PyTorch 2.5.1 | inference | No | CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.5.1-cpu-py311-ubuntu22.04-sagemaker |
343347
| PyTorch 2.5.1 | inference | No | GPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.5.1-gpu-py311-cu124-ubuntu22.04-sagemaker |
344348
| PyTorch 2.4.0 | training | No | CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.4.0-cpu-py311-ubuntu22.04-sagemaker |

0 commit comments

Comments
 (0)