runtime

Author: Borda

.lightning/workflows/fabric.yml

@@ -9,17 +9,17 @@ parametrize:
   matrix: {}
   include:
     # note that this is setting also all oldest requirements which is linked to python == 3.10
-    - image: "nvidia/cuda:12.1.1-devel-ubuntu22.04"
+    - image: "nvidia/cuda:12.1.1-runtime-ubuntu22.04"
       PACKAGE_NAME: "fabric"
       python_version: "3.10"
       machine: "A100_X_2"
-    - image: "nvidia/cuda:12.6.3-devel-ubuntu22.04"
+    - 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-devel-ubuntu22.04"
+    # - image: "nvidia/cuda:12.6-runtime-ubuntu22.04"
     #   PACKAGE_NAME: "fabric"
-    - image: "nvidia/cuda:12.6.3-devel-ubuntu22.04"
+    - image: "nvidia/cuda:12.6.3-runtime-ubuntu22.04"
       PACKAGE_NAME: "lightning"
       python_version: "3.12"
       machine: "L4_X_2"