bitsandbytes

Author: Borda

.azure/gpu-tests-fabric.yml

@@ -116,13 +116,6 @@ jobs:
           pip install setuptools==75.6.0 jsonargparse==4.35.0
         displayName: "Install package & dependencies"
 
-      - bash: |
-          set -e
-          python requirements/collect_env_details.py
-          python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu == 2, f'GPU: {mgpu}'"
-          python requirements/pytorch/check-avail-extras.py
-        displayName: "Env details"
-
       - bash: |
           # get pytorch version
           PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
@@ -134,6 +127,13 @@ jobs:
           fi
         displayName: "Handle bitsandbytes"
 
+      - bash: |
+          set -e
+          python requirements/collect_env_details.py
+          python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu == 2, f'GPU: {mgpu}'"
+          python requirements/pytorch/check-avail-extras.py
+        displayName: "Env details"
+
       - bash: python -m pytest lightning_fabric
         workingDirectory: src
         # without succeeded this could run even if the job has already failed

.azure/gpu-tests-pytorch.yml

@@ -130,13 +130,6 @@ jobs:
         condition: and(succeeded(), eq(variables['PACKAGE_NAME'], 'lightning'))
         displayName: "Drop PL for LAI"
 
-      - bash: |
-          set -e
-          python requirements/collect_env_details.py
-          python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu == 2, f'GPU: {mgpu}'"
-          python requirements/pytorch/check-avail-extras.py
-        displayName: "Env details"
-
       - bash: |
           # get pytorch version
           PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
@@ -148,6 +141,13 @@ jobs:
           fi
         displayName: "Handle bitsandbytes"
 
+      - bash: |
+          set -e
+          python requirements/collect_env_details.py
+          python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu == 2, f'GPU: {mgpu}'"
+          python requirements/pytorch/check-avail-extras.py
+        displayName: "Env details"
+
       - bash: python -m pytest pytorch_lightning
         workingDirectory: src
         # without succeeded this could run even if the job has already failed