From abe6362c68136a804dd1f238de6d74293fbd2563 Mon Sep 17 00:00:00 2001 From: NevermindNilas Date: Thu, 22 Aug 2024 21:47:24 +0300 Subject: [PATCH] add cuda toolkit --- .github/workflows/build.yaml | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 053c4d6e..93645789 100644 --- 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/cuda-toolkit@v0.2.16 + 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