check

Author: Borda

.azure/gpu-tests-fabric.yml

@@ -120,7 +120,7 @@ jobs:
           set -e
           python requirements/collect_env_details.py
           python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu == 2, f'GPU: {mgpu}'"
-          python -c "import bitsandbytes"
+          python requirements/pytorch/check-avail-extras.py
         displayName: "Env details"
 
       - bash: python -m pytest lightning_fabric

.azure/gpu-tests-pytorch.yml

@@ -135,7 +135,6 @@ jobs:
           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
-          python -c "import bitsandbytes"
         displayName: "Env details"
 
       - bash: python -m pytest pytorch_lightning

requirements/pytorch/check-avail-extras.py

@@ -4,3 +4,7 @@
     import matplotlib  # noqa: F401
     import omegaconf  # noqa: F401
     import rich  # noqa: F401
+
+    import torch  # noqa: F401
+    if torch.cuda.is_available():
+        import bitsandbytes  # noqa: F401