Skip to content

Commit 3b8d216

Browse files
committed
ci: install versioned CUDA toolkit
1 parent 24af20a commit 3b8d216

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

.github/workflows/build-wheels-cuda.yaml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,8 @@ jobs:
9999
MAMBA_NO_LOW_SPEED_LIMIT: "1"
100100
run: |
101101
$cudaVersion = $env:CUDAVER
102-
mamba install -y 'cuda' -c nvidia/label/cuda-$cudaVersion
102+
$cudaSeries = ($cudaVersion.Split('.')[0..1] -join '-')
103+
mamba install -y "cuda-toolkit-$cudaSeries" -c nvidia/label/cuda-$cudaVersion
103104
python -m pip install build wheel
104105
105106
- name: Build Wheel

0 commit comments

Comments
 (0)