Skip to content

Commit efacbc5

Browse files
committed
Use Torch 2.0.1 for cu117
1 parent f40e837 commit efacbc5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/build-wheels-release.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
name: Build Wheels & Release
1+
pip name: Build Wheels & Release
22

33
on:
44
workflow_dispatch:
@@ -71,7 +71,7 @@ jobs:
7171
if (!(mamba list cuda)[-1].contains('cuda')) {sleep -s 10; mamba install -y 'cuda' $cudaChannels.TrimEnd().Split()}
7272
if (!(mamba list cuda)[-1].contains('cuda')) {throw 'CUDA Toolkit failed to install!'}
7373
74-
if ([version]$env:CUDAVER -lt [version]'11.8.0') {$torchver = "torch==2.2.0"} else {$torchver = "torch==2.2.0"}
74+
if ([version]$env:CUDAVER -lt [version]'11.8.0') {$torchver = "torch==2.0.1"} else {$torchver = "torch==2.2.0"}
7575
python -m pip install $torchver --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
7676
7777
python -m pip install --upgrade setuptools==65.5.1

0 commit comments

Comments
 (0)