Skip to content

Commit b285eca

Browse files
JyothirmaikottuJyothirmaikottu
authored
Add 2.7.1 PT images to available images (#4918)
* added 2.7.1 PT images * cu128 * move 2.6 PT training to prior releaseS --------- Co-authored-by: jkottu <jkottu@amazon.com>
1 parent 18219e0 commit b285eca

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

‎available_images.md‎

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -120,8 +120,8 @@ EC2 Framework Containers (EC2, ECS, and EKS support only)
120120

121121
| Framework |Job Type |Horovod Options|CPU/GPU |Python Version Options | Example URL |
122122
|-------------------|-----------|---------------|-----------|-----------------------|-----------------------------------------------------------------------------------------------------------|
123-
| 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 |
124-
| 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 |
123+
| PyTorch 2.7.1 |training |No |CPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.7.1-cpu-py312-ubuntu22.04-ec2 |
124+
| PyTorch 2.7.1 |training |No |GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.7.1-gpu-py312-cu128-ubuntu22.04-ec2 |
125125
| 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 |
126126
| 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 |
127127
| 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 |
@@ -135,8 +135,8 @@ SageMaker Framework Containers (SM support only)
135135

136136
| Framework | Job Type | Horovod Options | CPU/GPU | Python Version Options | Example URL |
137137
|-------------------|-----------|-----------------|-----------|------------------------|----------------------------------------------------------------------------------------------------------------|
138-
| 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 |
139-
| 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 |
138+
| PyTorch 2.7.1 | training | No | CPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.7.1-cpu-py312-ubuntu22.04-sagemaker |
139+
| PyTorch 2.7.1 | training | No | GPU | 3.12 (py312) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.7.1-gpu-py312-cu128-ubuntu22.04-sagemaker |
140140
| 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 |
141141
| 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 |
142142
| 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 |
@@ -346,6 +346,8 @@ Prior EC2 Framework Container Versions
346346
==============
347347
| Framework |Job Type |Horovod Options |CPU/GPU |Python Version Options |Example URL |
348348
|---------------------------------------------|------------------------|---------------------------------|------------|---------------------------------------|----------------------------------------------------------------------------------------------------|
349+
| 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 |
350+
| 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 |
349351
| 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 |
350352
| 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 |
351353
| 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 |
@@ -359,6 +361,8 @@ Prior SageMaker Framework Container Versions
359361
==============
360362
| Framework |Job Type |Horovod Options |CPU/GPU |Python Version Options |Example URL |
361363
|---------------------------------------------|------------------------|---------------------------------|------------|---------------------------------------|----------------------------------------------------------------------------------------------------|
364+
| 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 |
365+
| 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 |
362366
| 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 |
363367
| 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 |
364368
| 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 |

0 commit comments

Comments
 (0)