CI: fix cuda-toolkit speed issue

bitsandbytes-foundation · Feb 8, 2024 · 3465f69 · 3465f69
1 parent 136721a
commit 3465f69
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -106,7 +106,9 @@ jobs:
       with:
         cuda: ${{ matrix.cuda_version }}
         method: 'local'
-        # sub-packages: '["nvcc","cudart","nvrtc_dev","cublas_dev","cusparse_dev","visual_studio_integration"]'
+        sub-packages: '["nvcc","cudart","cusparse","cublas","thrust","nvrtc_dev","cublas_dev","cusparse_dev"]'
+        linux-local-args: '["--toolkit"]'
+        use-github-cache: false
     - name: Add msbuild to PATH
       uses: microsoft/[email protected]
       if: ${{ startsWith(matrix.os, 'windows') }}