Update CUDA sub-packages in build-project.yaml

locaal-ai · Mar 21, 2024 · fa192e3 · fa192e3
1 parent 36488a9
commit fa192e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/build-project.yaml b/.github/workflows/build-project.yaml
@@ -243,13 +243,13 @@ jobs:
         uses: Jimver/[email protected]
         with:
           cuda: '${{ matrix.cublas }}'
-          sub-packages: '["nvcc", "cudart"]'
-          non-cuda-sub-packages: '["libcublas"]'
+          sub-packages: '["nvcc", "cudart", "cublas"]'
           method: 'local'
 
       - name: Set CUDA_TOOLKIT_ROOT_DIR if CUDA is installed
         if: ${{ matrix.cublas != 'cpu' }}
         run: |
+          ls "$env:CUDA_PATH\bin"
           "CUDA_TOOLKIT_ROOT_DIR=$env:CUDA_PATH" >> $env:GITHUB_ENV
 
       - name: Set Up Environment 🔧