Skip to content

Commit

Permalink
ci(cuda): install CUDA toolkit to build extension
Browse files Browse the repository at this point in the history
  • Loading branch information
dacorvo committed Apr 10, 2024
1 parent 8ecc1d3 commit 3338c00
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions .github/workflows/linux-cuda-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,20 @@ jobs:

steps:
- uses: actions/checkout@v2
- uses: Jimver/[email protected]
id: cuda-toolkit
with:
cuda: '12.3.2'
- run: |
echo "Installed cuda version is: ${{steps.cuda-toolkit.outputs.cuda}}"
echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
nvcc -V
- name: Install CUDA Toolkit
shell: bash
run: |
wget --progress=bar:force:noscroll https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run
nvcc --version
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v2
with:
Expand Down

0 comments on commit 3338c00

Please sign in to comment.