Add sub-packages and non-cuda-sub-packages options to CUDA toolkit setup

locaal-ai · Mar 21, 2024 · 36488a9 · 36488a9
1 parent f2f8502
commit 36488a9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/build-project.yaml b/.github/workflows/build-project.yaml
@@ -243,6 +243,9 @@ jobs:
         uses: Jimver/[email protected]
         with:
           cuda: '${{ matrix.cublas }}'
+          sub-packages: '["nvcc", "cudart"]'
+          non-cuda-sub-packages: '["libcublas"]'
+          method: 'local'
 
       - name: Set CUDA_TOOLKIT_ROOT_DIR if CUDA is installed
         if: ${{ matrix.cublas != 'cpu' }}