From e22a4d128c097d4c16e7dac7ad26c743c9f0ca71 Mon Sep 17 00:00:00 2001
From: Yadan-Wei <84998528+Yadan-Wei@users.noreply.github.com>
Date: Fri, 24 Jan 2025 16:41:18 -0800
Subject: [PATCH] add tf2.18 training images to release (#4510)

Co-authored-by: Yadan Wei <yadanwei@amazon.com>
---
 release_images_patches.yml | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/release_images_patches.yml b/release_images_patches.yml
index a48c1dad4155..3cd3e2a5f6bf 100644
--- a/release_images_patches.yml
+++ b/release_images_patches.yml
@@ -368,7 +368,35 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
+  
+  # TensorFlow 2.18 Images
   21:
+    framework: "tensorflow"
+    version: "2.18.0"
+    customer_type: "ec2"
+    arch_type: "x86"
+    training:
+      device_types: [ "cpu", "gpu" ]
+      python_versions: [ "py310" ]
+      os_version: "ubuntu22.04"
+      cuda_version: "cu125"
+      example: False
+      disable_sm_tag: False
+      force_release: False
+  22:
+    framework: "tensorflow"
+    version: "2.18.0"
+    customer_type: "sagemaker"
+    arch_type: "x86"
+    training:
+      device_types: [ "cpu", "gpu" ]
+      python_versions: [ "py310" ]
+      os_version: "ubuntu22.04"
+      cuda_version: "cu125"
+      example: False
+      disable_sm_tag: False
+      force_release: False
+  23:
     framework: "tensorflow"
     version: "2.18.0"
     arch_type: "x86"
@@ -381,7 +409,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  22:
+  24:
     framework: "tensorflow"
     version: "2.18.0"
     arch_type: "x86"