Release PyTorch 2.4.0 Training DLC (#4273)

* Release PyTorch 2.4 Training EC2 DLC * add sm
aws · Sep 23, 2024 · 771cabc · 771cabc
1 parent ce5fb7a
commit 771cabc
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/release_images_training.yml b/release_images_training.yml
@@ -64,3 +64,29 @@ release_images:
       example: False
       disable_sm_tag: True
       force_release: False
+  6:
+    framework: "pytorch"
+    version: "2.4.0"
+    customer_type: "ec2"
+    arch_type: "x86"
+    training:
+      device_types: [ "cpu", "gpu" ]
+      python_versions: [ "py311" ]
+      os_version: "ubuntu22.04"
+      cuda_version: "cu124"
+      example: False
+      disable_sm_tag: False
+      force_release: False
+  7:
+    framework: "pytorch"
+    version: "2.4.0"
+    customer_type: "sagemaker"
+    arch_type: "x86"
+    training:
+      device_types: [ "cpu", "gpu" ]
+      python_versions: [ "py311" ]
+      os_version: "ubuntu22.04"
+      cuda_version: "cu124"
+      example: False
+      disable_sm_tag: False
+      force_release: False