Release PyTorch 2.1 Training Autopatch (#3855)

Co-authored-by: Sirut Buasai <[email protected]>
aws · May 9, 2024 · bc13cd9 · bc13cd9
1 parent 1bb4b09
commit bc13cd9
Showing 1 changed file with 32 additions and 19 deletions.
diff --git a/release_images_training.yml b/release_images_training.yml
@@ -40,19 +40,6 @@ release_images:
       disable_sm_tag: False
       force_release: False
   4:
-    framework: "pytorch"
-    version: "2.1.0"
-    customer_type: "sagemaker"
-    arch_type: "x86"
-    training:
-      device_types: [ "cpu", "gpu" ]
-      python_versions: [ "py310" ]
-      os_version: "ubuntu20.04"
-      cuda_version: "cu121"
-      example: False
-      disable_sm_tag: False
-      force_release: False
-  5:
     framework: "pytorch"
     version: "2.2.0"
     arch_type: "x86"
@@ -65,7 +52,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  6:
+  5:
     framework: "pytorch"
     version: "2.0.1"
     customer_type: "sagemaker"
@@ -78,7 +65,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  7:
+  6:
     framework: "huggingface_pytorch"
     version: "1.13.1"
     arch_type: "x86"
@@ -91,7 +78,7 @@ release_images:
       example: False
       disable_sm_tag: True
       force_release: False
-  8:
+  7:
     framework: "huggingface_pytorch"
     version: "1.13.1"
     arch_type: "x86"
@@ -104,7 +91,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  9:
+  8:
     framework: "tensorflow"
     version: "2.14.1"
     customer_type: "ec2"
@@ -117,7 +104,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  10:
+  9:
     framework: "tensorflow"
     version: "2.14.1"
     customer_type: "sagemaker"
@@ -130,7 +117,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  11:
+  10:
     framework: "tensorflow"
     version: "2.13.0"
     customer_type: "ec2"
@@ -143,3 +130,29 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
+  11:
+    framework: "pytorch"
+    version: "2.1.0"
+    customer_type: "ec2"
+    arch_type: "x86"
+    training:
+      device_types: [ "cpu", "gpu" ]
+      python_versions: [ "py310" ]
+      os_version: "ubuntu20.04"
+      cuda_version: "cu121"
+      example: False
+      disable_sm_tag: False
+      force_release: False
+  12:
+    framework: "pytorch"
+    version: "2.1.0"
+    customer_type: "sagemaker"
+    arch_type: "x86"
+    training:
+      device_types: [ "cpu", "gpu" ]
+      python_versions: [ "py310" ]
+      os_version: "ubuntu20.04"
+      cuda_version: "cu121"
+      example: False
+      disable_sm_tag: False
+      force_release: False