add tf2.18 training images to release (#4510)

Co-authored-by: Yadan Wei <[email protected]>
aws · Jan 25, 2025 · e22a4d1 · e22a4d1
1 parent 01e32fc
commit e22a4d1
Showing 1 changed file with 29 additions and 1 deletion.
diff --git 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"