We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f40e837 commit efacbc5Copy full SHA for efacbc5
.github/workflows/build-wheels-release.yml
@@ -1,4 +1,4 @@
1
-name: Build Wheels & Release
+pip name: Build Wheels & Release
2
3
on:
4
workflow_dispatch:
@@ -71,7 +71,7 @@ jobs:
71
if (!(mamba list cuda)[-1].contains('cuda')) {sleep -s 10; mamba install -y 'cuda' $cudaChannels.TrimEnd().Split()}
72
if (!(mamba list cuda)[-1].contains('cuda')) {throw 'CUDA Toolkit failed to install!'}
73
74
- if ([version]$env:CUDAVER -lt [version]'11.8.0') {$torchver = "torch==2.2.0"} else {$torchver = "torch==2.2.0"}
+ if ([version]$env:CUDAVER -lt [version]'11.8.0') {$torchver = "torch==2.0.1"} else {$torchver = "torch==2.2.0"}
75
python -m pip install $torchver --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
76
77
python -m pip install --upgrade setuptools==65.5.1
0 commit comments