add cuda toolkit

NevermindNilas · Aug 22, 2024 · abe6362 · abe6362
1 parent f597a14
commit abe6362
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -58,6 +58,20 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
+      - name: Install CUDA Toolkit (Ubuntu)
+        if: matrix.os == 'ubuntu-latest'
+        uses: Jimver/[email protected]
+        id: cuda-toolkit
+        with:
+          cuda: '12.4.0'
+
+      - name: Verify CUDA Installation (Ubuntu)
+        if: matrix.os == 'ubuntu-latest'
+        run: |
+          echo "Installed cuda version is: ${{ steps.cuda-toolkit.outputs.cuda }}"
+          echo "Cuda install location: ${{ steps.cuda-toolkit.outputs.CUDA_PATH }}"
+          nvcc -V
+
       - name: Sanity Check - List pip modules
         run: python -m pip list