Revert "image"

This reverts commit 1037779.

Author: Borda

.github/checkgroup.yml

@@ -47,9 +47,9 @@ subprojects:
       - "!*.md"
       - "!**/*.md"
     checks:
-      - "pytorch.yml / Lit Job (pytorch, 3.10, L4_X_2)"
-      - "pytorch.yml / Lit Job (lightning, 3.12, L4_X_2)"
-      - "pytorch.yml / Lit Job (pytorch, 3.12, L4_X_2)"
+      - "pytorch.yml / Lit Job (nvidia/cuda:12.1.1-runtime-ubuntu22.04, pytorch, 3.10, L4_X_2)"
+      - "pytorch.yml / Lit Job (nvidia/cuda:12.6.3-runtime-ubuntu22.04, lightning, 3.12, L4_X_2)"
+      - "pytorch.yml / Lit Job (nvidia/cuda:12.6.3-runtime-ubuntu22.04, pytorch, 3.12, L4_X_2)"
 
   - id: "Benchmarks"
     paths:
@@ -148,9 +148,9 @@ subprojects:
       - "!*.md"
       - "!**/*.md"
     checks:
-      - "fabric.yml / Lit Job (fabric, 3.10, L4_X_2)"
-      - "fabric.yml / Lit Job (fabric, 3.12, L4_X_2)"
-      - "fabric.yml / Lit Job (lightning, 3.12, L4_X_2)"
+      - "fabric.yml / Lit Job (nvidia/cuda:12.1.1-runtime-ubuntu22.04, fabric, 3.10, L4_X_2)"
+      - "fabric.yml / Lit Job (nvidia/cuda:12.6.3-runtime-ubuntu22.04, fabric, 3.12, L4_X_2)"
+      - "fabric.yml / Lit Job (nvidia/cuda:12.6.3-runtime-ubuntu22.04, lightning, 3.12, L4_X_2)"
 
   # Temporarily disabled
   #  - id: "lightning_fabric: TPU workflow"

.lightning/workflows/fabric.yml

@@ -5,20 +5,22 @@ trigger:
     branches: ["master", "release/stable"]
 
 timeout: "55" # minutes
-image: "nvidia/cuda:12.6.3-runtime-ubuntu22.04"
 parametrize:
   matrix: {}
   include:
     # note that this is setting also all oldest requirements which is linked to python == 3.10
-    - PACKAGE_NAME: "fabric"
+    - image: "nvidia/cuda:12.1.1-runtime-ubuntu22.04"
+      PACKAGE_NAME: "fabric"
       python_version: "3.10"
       machine: "L4_X_2"
-    - PACKAGE_NAME: "fabric"
+    - image: "nvidia/cuda:12.6.3-runtime-ubuntu22.04"
+      PACKAGE_NAME: "fabric"
       python_version: "3.12"
       machine: "L4_X_2"
     # - image: "nvidia/cuda:12.6-runtime-ubuntu22.04"
     #   PACKAGE_NAME: "fabric"
-    - PACKAGE_NAME: "lightning"
+    - image: "nvidia/cuda:12.6.3-runtime-ubuntu22.04"
+      PACKAGE_NAME: "lightning"
       python_version: "3.12"
       machine: "L4_X_2"
   exclude: []

.lightning/workflows/pytorch.yml

@@ -5,20 +5,22 @@ trigger:
     branches: ["master", "release/stable"]
 
 timeout: "55" # minutes
-image: "nvidia/cuda:12.6.3-runtime-ubuntu22.04"
 parametrize:
   matrix: {}
   include:
     # note that this also sets oldest requirements which are linked to Python == 3.10
-    - PACKAGE_NAME: "pytorch"
+    - image: "nvidia/cuda:12.1.1-runtime-ubuntu22.04"
+      PACKAGE_NAME: "pytorch"
       python_version: "3.10"
       machine: "L4_X_2"
-    - PACKAGE_NAME: "pytorch"
+    - image: "nvidia/cuda:12.6.3-runtime-ubuntu22.04"
+      PACKAGE_NAME: "pytorch"
       python_version: "3.12"
       machine: "L4_X_2"
     # - image: "nvidia/cuda:12.6.3-runtime-ubuntu22.04"
     #   PACKAGE_NAME: "pytorch"
-    - PACKAGE_NAME: "lightning"
+    - image: "nvidia/cuda:12.6.3-runtime-ubuntu22.04"
+      PACKAGE_NAME: "lightning"
       python_version: "3.12"
       machine: "L4_X_2"
   exclude: []