From 02da6c961ebe8477835c37e565ef76d8350a5a6c Mon Sep 17 00:00:00 2001 From: Yadan Wei Date: Tue, 18 Feb 2025 13:27:01 -0800 Subject: [PATCH] add PT2.6 x86 Inf EC2 to release --- release_images_patches.yml | 37 +++++++++++++++++++++++++------------ 1 file changed, 25 insertions(+), 12 deletions(-) diff --git a/release_images_patches.yml b/release_images_patches.yml index 3a0f5bd79a5d..d7ae21f22a6f 100644 --- a/release_images_patches.yml +++ b/release_images_patches.yml @@ -239,6 +239,19 @@ release_images: # PyTorch 2.6 Images 19: + framework: "pytorch" + version: "2.6.0" + arch_type: "x86" + customer_type: "ec2" + inference: + device_types: [ "cpu", "gpu" ] + python_versions: [ "py312" ] + os_version: "ubuntu22.04" + cuda_version: "cu124" + example: False + disable_sm_tag: False + force_release: False + 20: framework: "pytorch" version: "2.6.0" arch_type: "x86" @@ -253,7 +266,7 @@ release_images: force_release: False # TensorFlow 2.16 Images - 20: + 21: framework: "tensorflow" version: "2.16.2" customer_type: "ec2" @@ -266,7 +279,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 21: + 22: framework: "tensorflow" version: "2.16.2" customer_type: "sagemaker" @@ -279,7 +292,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 22: + 23: framework: "tensorflow" version: "2.16.1" arch_type: "x86" @@ -292,7 +305,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 23: + 24: framework: "tensorflow" version: "2.16.1" arch_type: "x86" @@ -305,7 +318,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 24: + 25: framework: "tensorflow" version: "2.16.1" arch_type: "graviton" @@ -318,7 +331,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 25: + 26: framework: "tensorflow" version: "2.16.1" arch_type: "graviton" @@ -333,7 +346,7 @@ release_images: force_release: False # TensorFlow 2.18 Images - 26: + 27: framework: "tensorflow" version: "2.18.0" customer_type: "ec2" @@ -346,7 +359,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 27: + 28: framework: "tensorflow" version: "2.18.0" customer_type: "sagemaker" @@ -359,7 +372,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 27: + 29: framework: "tensorflow" version: "2.18.0" arch_type: "x86" @@ -372,7 +385,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 28: + 30: framework: "tensorflow" version: "2.18.0" arch_type: "x86" @@ -385,7 +398,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 29: + 31: framework: "tensorflow" version: "2.18.0" arch_type: "arm64" @@ -398,7 +411,7 @@ release_images: example: False disable_sm_tag: False force_release: False - 30: + 32: framework: "tensorflow" version: "2.18.0" arch_type: "arm64"