Actions: Fix torch URL

We only build wheels for cuda 12 Signed-off-by: kingbri <bdashore3@proton.me>
bdashore3 · Feb 2, 2024 · e4f2b9b · e4f2b9b
1 parent 0280f3e
commit e4f2b9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/build-wheels.yml b/.github/workflows/build-wheels.yml
@@ -68,7 +68,7 @@ jobs:
           if (!(mamba list cuda)[-1].contains('cuda')) {sleep -s 10; mamba install -y 'cuda' $cudaChannels.TrimEnd().Split()}
           if (!(mamba list cuda)[-1].contains('cuda')) {throw 'CUDA Toolkit failed to install!'}
           
-          python -m pip install --upgrade build setuptools wheel packaging ninja torch==${{ matrix.torchver }} --extra-index-url "https://download.pytorch.org/whl/cu$cudaVersionPytorch"
+          python -m pip install --upgrade build setuptools wheel packaging ninja torch==${{ matrix.torchver }} --extra-index-url "https://download.pytorch.org/whl/cu121"
 
       - name: Build Wheel
         id: build-wheel