bitsandbytes

Author: Borda

.azure/gpu-tests-fabric.yml

@@ -119,11 +119,8 @@ 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
-        displayName: "Env details"
-
-      - bash: |
           python -c "import bitsandbytes"
-        displayName: "Handle bitsandbytes"
+        displayName: "Env details"
 
       - bash: python -m pytest lightning_fabric
         workingDirectory: src

.azure/gpu-tests-pytorch.yml

@@ -133,11 +133,8 @@ 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
-        displayName: "Env details"
-
-      - bash: |
           python -c "import bitsandbytes"
-        displayName: "Handle bitsandbytes"
+        displayName: "Env details"
 
       - bash: python -m pytest pytorch_lightning
         workingDirectory: src

requirements/pytorch/extra.txt

@@ -8,5 +8,4 @@ hydra-core >=1.2.0, <1.4.0
 jsonargparse[signatures] >=4.27.7, <=4.35.0
 rich >=12.3.0, <13.6.0
 tensorboardX >=2.2, <2.7.0  # min version is set by torch.onnx missing attribute
-bitsandbytes >=0.44.0,<0.45.3; sys_platform == 'linux' or sys_platform == 'win32'
-bitsandbytes >=0.42.0,<0.45.3 ; sys_platform == 'darwin'
+bitsandbytes >=0.45.2,<0.45.3; platform_system != "Darwin"