Skip to content

Commit 6a9d525

Browse files
authored
add pytorch 2.8 training images to release patches (#5226)
1 parent fe89ca3 commit 6a9d525

File tree

1 file changed

+40
-9
lines changed

1 file changed

+40
-9
lines changed

release_images_patches.yml

Lines changed: 40 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -214,8 +214,39 @@ release_images:
214214
disable_sm_tag: False
215215
force_release: False
216216
public_registry: True
217-
# TensorFlow 2.18 Images
217+
218+
# PyTorch 2.8 Images
218219
16:
220+
framework: "pytorch"
221+
version: "2.8.0"
222+
customer_type: "ec2"
223+
arch_type: "x86"
224+
training:
225+
device_types: ["cpu", "gpu"]
226+
python_versions: [ "py312" ]
227+
os_version: "ubuntu22.04"
228+
cuda_version: "cu129"
229+
example: False
230+
disable_sm_tag: False
231+
force_release: False
232+
public_registry: True
233+
17:
234+
framework: "pytorch"
235+
version: "2.8.0"
236+
customer_type: "sagemaker"
237+
arch_type: "x86"
238+
training:
239+
device_types: ["cpu", "gpu"]
240+
python_versions: [ "py312" ]
241+
os_version: "ubuntu22.04"
242+
cuda_version: "cu129"
243+
example: False
244+
disable_sm_tag: False
245+
force_release: False
246+
public_registry: True
247+
248+
# TensorFlow 2.18 Images
249+
18:
219250
framework: "tensorflow"
220251
version: "2.18.0"
221252
customer_type: "ec2"
@@ -229,7 +260,7 @@ release_images:
229260
disable_sm_tag: False
230261
force_release: False
231262
public_registry: True
232-
17:
263+
19:
233264
framework: "tensorflow"
234265
version: "2.18.0"
235266
customer_type: "sagemaker"
@@ -243,7 +274,7 @@ release_images:
243274
disable_sm_tag: False
244275
force_release: False
245276
public_registry: True
246-
18:
277+
20:
247278
framework: "tensorflow"
248279
version: "2.18.0"
249280
arch_type: "x86"
@@ -257,7 +288,7 @@ release_images:
257288
disable_sm_tag: False
258289
force_release: False
259290
public_registry: True
260-
19:
291+
21:
261292
framework: "tensorflow"
262293
version: "2.18.0"
263294
arch_type: "x86"
@@ -271,7 +302,7 @@ release_images:
271302
disable_sm_tag: False
272303
force_release: False
273304
public_registry: True
274-
20:
305+
22:
275306
framework: "tensorflow"
276307
version: "2.18.0"
277308
arch_type: "arm64"
@@ -285,7 +316,7 @@ release_images:
285316
disable_sm_tag: False
286317
force_release: False
287318
public_registry: True
288-
21:
319+
23:
289320
framework: "tensorflow"
290321
version: "2.18.0"
291322
arch_type: "arm64"
@@ -300,7 +331,7 @@ release_images:
300331
force_release: False
301332
public_registry: True
302333
# TensorFlow 2.19 Images
303-
22:
334+
24:
304335
framework: "tensorflow"
305336
version: "2.19.0"
306337
customer_type: "sagemaker"
@@ -314,7 +345,7 @@ release_images:
314345
disable_sm_tag: False
315346
force_release: False
316347
public_registry: True
317-
23:
348+
25:
318349
framework: "tensorflow"
319350
version: "2.19.0"
320351
arch_type: "x86"
@@ -328,7 +359,7 @@ release_images:
328359
disable_sm_tag: False
329360
force_release: False
330361
public_registry: True
331-
24:
362+
26:
332363
framework: "tensorflow"
333364
version: "2.19.0"
334365
arch_type: "arm64"

0 commit comments

Comments
 (0)