We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 24af20a commit 3b8d216Copy full SHA for 3b8d216
1 file changed
.github/workflows/build-wheels-cuda.yaml
@@ -99,7 +99,8 @@ jobs:
99
MAMBA_NO_LOW_SPEED_LIMIT: "1"
100
run: |
101
$cudaVersion = $env:CUDAVER
102
- mamba install -y 'cuda' -c nvidia/label/cuda-$cudaVersion
+ $cudaSeries = ($cudaVersion.Split('.')[0..1] -join '-')
103
+ mamba install -y "cuda-toolkit-$cudaSeries" -c nvidia/label/cuda-$cudaVersion
104
python -m pip install build wheel
105
106
- name: Build Wheel
0 commit comments