ci / gpu

Borda · Borda · commit 60dddec875d2 · 2025-03-18T23:09:24.000+01:00
diff --git a/.azure/gpu-tests-fabric.yml b/.azure/gpu-tests-fabric.yml
@@ -62,6 +62,9 @@ jobs:
         "Fabric | latest":
           image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
           PACKAGE_NAME: "fabric"
+        "Fabric | future":
+          image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.7-cuda12.6.3"
+          PACKAGE_NAME: "fabric"
         "Lightning | latest":
           image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
           PACKAGE_NAME: "lightning"
diff --git a/.azure/gpu-tests-pytorch.yml b/.azure/gpu-tests-pytorch.yml
@@ -55,6 +55,9 @@ jobs:
         "PyTorch | latest":
           image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
           PACKAGE_NAME: "pytorch"
+        "PyTorch | future":
+          image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.7-cuda12.6.3"
+          PACKAGE_NAME: "pytorch"
         "Lightning | latest":
           image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
           PACKAGE_NAME: "lightning"
