gpu

Author: Borda

.azure/gpu-tests-fabric.yml

@@ -56,8 +56,11 @@ jobs:
       options: "--gpus=all --shm-size=2gb  -v /var/tmp:/var/tmp"
     strategy:
       matrix:
+        "Fabric | oldest":
+          image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.1"
+          PACKAGE_NAME: "fabric"
         "Fabric | latest":
-          image: "pytorchlightning/pytorch_lightning:base-cuda-py3.11-torch2.3-cuda12.1.1"
+          image: "pytorchlightning/pytorch_lightning:base-cuda-py3.11-torch2.5-cuda12.1.1"
           PACKAGE_NAME: "fabric"
         "Lightning | latest":
           image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.5-cuda12.1.1"

.azure/gpu-tests-pytorch.yml

@@ -49,8 +49,11 @@ jobs:
     cancelTimeoutInMinutes: "2"
     strategy:
       matrix:
+        "PyTorch | oldest":
+          image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.1"
+          PACKAGE_NAME: "pytorch"
         "PyTorch | latest":
-          image: "pytorchlightning/pytorch_lightning:base-cuda-py3.11-torch2.3-cuda12.1.1"
+          image: "pytorchlightning/pytorch_lightning:base-cuda-py3.11-torch2.5-cuda12.1.1"
           PACKAGE_NAME: "pytorch"
         "Lightning | latest":
           image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.5-cuda12.1.1"