Skip to content

Commit 2e7c808

Browse files
bhanutejagkBhanu Teja Goshikonda
bhanutejagk
and
Bhanu Teja Goshikonda
authored
Depricate PyTorch 2.5 (#5440)
* Depricate PyTorch 2.5 cr: https://code.amazon.com/reviews/CR-232265723 * reverted back deletions for huggingface and neuron containers for deprecating pytorch 2.5.1 --------- Co-authored-by: Bhanu Teja Goshikonda <[email protected]>
1 parent 7c39d33 commit 2e7c808

File tree

2 files changed

+14
-120
lines changed

2 files changed

+14
-120
lines changed

‎available_images.md‎

Lines changed: 0 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -387,10 +387,6 @@ Prior EC2 Framework Container Versions
387387
| 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 |
388388
| 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 |
389389
| 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 |
390-
| 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 |
391-
| 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 |
392-
| 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 |
393-
| 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 |
394390

395391
Prior SageMaker Framework Container Versions
396392
==============
@@ -400,23 +396,6 @@ Prior SageMaker Framework Container Versions
400396
| 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 |
401397
| 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 |
402398
| 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 |
403-
| 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 |
404-
| 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 |
405-
| 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 |
406-
| 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 |
407-
408-
Prior EC2 Framework ARM64/Graviton Containers
409-
============================
410-
| Framework |Job Type |Horovod Options|CPU/GPU |Python Version Options |Example URL |
411-
|-------------------|-----------|---------------|-----------|-----------------------|---------------------------------------------------------------------------------------------------|
412-
| PyTorch 2.5.1 |inference |No |CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-arm64:2.5.1-cpu-py311-ubuntu22.04-ec2 |
413-
| PyTorch 2.5.1 |inference |No |GPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-arm64:2.5.1-gpu-py311-cu124-ubuntu22.04-ec2 |
414-
415-
Prior SageMaker Framework ARM64/Graviton Containers
416-
============================
417-
| Framework |Job Type |Horovod Options|CPU/GPU |Python Version Options |Example URL |
418-
|-------------------|-----------|---------------|-----------|-----------------------|---------------------------------------------------------------------------------------------------|
419-
| PyTorch 2.5.1 | inference | No | CPU | 3.11 (py311) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-arm64:2.5.1-cpu-py311-ubuntu22.04-sagemaker |
420399

421400
Prior AutoGluon Training Containers
422401
===============================

‎release_images_patches.yml‎

Lines changed: 14 additions & 99 deletions
Original file line numberDiff line numberDiff line change
@@ -1,93 +1,8 @@
11
---
22
# DLC images to patch - force_release is not allowed in this file
33
release_images:
4-
# PyTorch 2.5 Images
5-
1:
6-
framework: "pytorch"
7-
version: "2.5.1"
8-
customer_type: "ec2"
9-
arch_type: "x86"
10-
training:
11-
device_types: [ "cpu", "gpu" ]
12-
python_versions: [ "py311" ]
13-
os_version: "ubuntu22.04"
14-
cuda_version: "cu124"
15-
example: False
16-
disable_sm_tag: False
17-
force_release: False
18-
public_registry: True
19-
2:
20-
framework: "pytorch"
21-
version: "2.5.1"
22-
customer_type: "sagemaker"
23-
arch_type: "x86"
24-
training:
25-
device_types: [ "cpu", "gpu" ]
26-
python_versions: [ "py311" ]
27-
os_version: "ubuntu22.04"
28-
cuda_version: "cu124"
29-
example: False
30-
disable_sm_tag: False
31-
force_release: False
32-
public_registry: True
33-
3:
34-
framework: "pytorch"
35-
version: "2.5.1"
36-
arch_type: "x86"
37-
customer_type: "ec2"
38-
inference:
39-
device_types: [ "cpu", "gpu" ]
40-
python_versions: [ "py311" ]
41-
os_version: "ubuntu22.04"
42-
cuda_version: "cu124"
43-
example: False
44-
disable_sm_tag: False
45-
force_release: False
46-
public_registry: True
47-
4:
48-
framework: "pytorch"
49-
version: "2.5.1"
50-
arch_type: "x86"
51-
customer_type: "sagemaker"
52-
inference:
53-
device_types: [ "cpu", "gpu" ]
54-
python_versions: [ "py311" ]
55-
os_version: "ubuntu22.04"
56-
cuda_version: "cu124"
57-
example: False
58-
disable_sm_tag: False
59-
force_release: False
60-
public_registry: True
61-
5:
62-
framework: "pytorch"
63-
version: "2.5.1"
64-
arch_type: "arm64"
65-
customer_type: "ec2"
66-
inference:
67-
device_types: [ "cpu", "gpu" ]
68-
python_versions: [ "py311" ]
69-
os_version: "ubuntu22.04"
70-
cuda_version: "cu124"
71-
example: False
72-
disable_sm_tag: False
73-
force_release: False
74-
public_registry: True
75-
6:
76-
framework: "pytorch"
77-
version: "2.5.1"
78-
arch_type: "arm64"
79-
customer_type: "sagemaker"
80-
inference:
81-
device_types: [ "cpu" ]
82-
python_versions: [ "py311" ]
83-
os_version: "ubuntu22.04"
84-
example: False
85-
disable_sm_tag: False
86-
force_release: False
87-
public_registry: True
88-
894
# PyTorch 2.6 Images
90-
7:
5+
1:
916
framework: "pytorch"
927
version: "2.6.0"
938
customer_type: "ec2"
@@ -101,7 +16,7 @@ release_images:
10116
disable_sm_tag: False
10217
force_release: False
10318
public_registry: True
104-
8:
19+
2:
10520
framework: "pytorch"
10621
version: "2.6.0"
10722
customer_type: "sagemaker"
@@ -115,7 +30,7 @@ release_images:
11530
disable_sm_tag: False
11631
force_release: False
11732
public_registry: True
118-
9:
33+
3:
11934
framework: "pytorch"
12035
version: "2.6.0"
12136
arch_type: "x86"
@@ -129,7 +44,7 @@ release_images:
12944
disable_sm_tag: False
13045
force_release: False
13146
public_registry: True
132-
10:
47+
4:
13348
framework: "pytorch"
13449
version: "2.6.0"
13550
arch_type: "x86"
@@ -143,7 +58,7 @@ release_images:
14358
disable_sm_tag: False
14459
force_release: False
14560
public_registry: True
146-
11:
61+
5:
14762
framework: "pytorch"
14863
version: "2.6.0"
14964
arch_type: "arm64"
@@ -157,7 +72,7 @@ release_images:
15772
disable_sm_tag: False
15873
force_release: False
15974
public_registry: True
160-
12:
75+
6:
16176
framework: "pytorch"
16277
version: "2.6.0"
16378
arch_type: "arm64"
@@ -172,7 +87,7 @@ release_images:
17287
public_registry: True
17388

17489
# PyTorch 2.7 Images
175-
13:
90+
7:
17691
framework: "pytorch"
17792
version: "2.7.0"
17893
customer_type: "ec2"
@@ -187,7 +102,7 @@ release_images:
187102
force_release: False
188103
public_registry: True
189104
enable_soci: True
190-
14:
105+
8:
191106
framework: "pytorch"
192107
version: "2.7.1"
193108
customer_type: "ec2"
@@ -202,7 +117,7 @@ release_images:
202117
force_release: False
203118
public_registry: True
204119
enable_soci: True
205-
15:
120+
9:
206121
framework: "pytorch"
207122
version: "2.7.1"
208123
customer_type: "sagemaker"
@@ -219,7 +134,7 @@ release_images:
219134
enable_soci: True
220135

221136
# PyTorch 2.8 Images
222-
16:
137+
10:
223138
framework: "pytorch"
224139
version: "2.8.0"
225140
customer_type: "ec2"
@@ -234,7 +149,7 @@ release_images:
234149
force_release: False
235150
public_registry: True
236151
enable_soci: True
237-
17:
152+
11:
238153
framework: "pytorch"
239154
version: "2.8.0"
240155
customer_type: "sagemaker"
@@ -251,7 +166,7 @@ release_images:
251166
enable_soci: True
252167

253168
# TensorFlow 2.19 Images
254-
18:
169+
12:
255170
framework: "tensorflow"
256171
version: "2.19.0"
257172
customer_type: "sagemaker"
@@ -266,7 +181,7 @@ release_images:
266181
force_release: False
267182
public_registry: True
268183
enable_soci: True
269-
19:
184+
13:
270185
framework: "tensorflow"
271186
version: "2.19.0"
272187
arch_type: "x86"
@@ -280,7 +195,7 @@ release_images:
280195
disable_sm_tag: False
281196
force_release: False
282197
public_registry: True
283-
20:
198+
14:
284199
framework: "tensorflow"
285200
version: "2.19.0"
286201
arch_type: "arm64"

0 commit comments

Comments
 (0)