diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml index 280cffa..3e95559 100644 --- a/.github/workflows/wheels.yml +++ b/.github/workflows/wheels.yml @@ -40,6 +40,13 @@ jobs: os: [ubuntu-latest, windows-latest] steps: + - name: Fetch CUDA toolkit + uses: Jimver/cuda-toolkit@v0.2.11 + id: cuda-toolkit + with: + cuda: '12.5.0' + linux-local-args: '["--toolkit"]' + - uses: actions/checkout@v4 with: submodules: true