A100 & L4

Author: Borda

.lightning/workflows/fabric.yml

@@ -5,13 +5,14 @@ trigger:
     branches: ["master"]
 
 timeout: "55" # minutes
-machine: "A100_X_2"
+machine: "L4_X_2"
 parametrize:
   matrix: {}
   include:
     # note that this is setting also all oldest requirements which is linked to Torch == 2.1
     - image: "pytorchlightning/pytorch_lightning:base-cuda12.1.1-py3.10-torch2.1"
       PACKAGE_NAME: "fabric"
+      machine: "A100_X_2"
     - image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.8"
       PACKAGE_NAME: "fabric"
     # - image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.7"

.lightning/workflows/pytorch.yml

@@ -5,13 +5,14 @@ trigger:
     branches: ["master"]
 
 timeout: "55" # minutes
-machine: "A100_X_2"
+machine: "l4_X_2"
 parametrize:
   matrix: {}
   include:
     # note that this is setting also all oldest requirements which is linked to Torch == 2.1
     - image: "pytorchlightning/pytorch_lightning:base-cuda12.1.1-py3.10-torch2.1"
       PACKAGE_NAME: "pytorch"
+      machine: "A100_X_2"
     - image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.8"
       PACKAGE_NAME: "pytorch"
     # - image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.7"