Skip to content

Commit ca47f53

Browse files
committed
Actions: Fix torch URL
We only build wheels for cuda 12 Signed-off-by: kingbri <bdashore3@proton.me>
1 parent 3d56efd commit ca47f53

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/build-wheels.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ jobs:
6868
if (!(mamba list cuda)[-1].contains('cuda')) {sleep -s 10; mamba install -y 'cuda' $cudaChannels.TrimEnd().Split()}
6969
if (!(mamba list cuda)[-1].contains('cuda')) {throw 'CUDA Toolkit failed to install!'}
7070
71-
python -m pip install --upgrade build setuptools wheel packaging ninja torch==${{ matrix.torchver }} --extra-index-url "https://download.pytorch.org/whl/cu$cudaVersionPytorch"
71+
python -m pip install --upgrade build setuptools wheel packaging ninja torch==${{ matrix.torchver }} --extra-index-url "https://download.pytorch.org/whl/cu121"
7272
7373
- name: Build Wheel
7474
id: build-wheel

0 commit comments

Comments
 (0)