From e4f2b9b3befe25bde890189e577cb5f31bbbdda9 Mon Sep 17 00:00:00 2001 From: kingbri Date: Fri, 2 Feb 2024 14:31:06 -0500 Subject: [PATCH] Actions: Fix torch URL We only build wheels for cuda 12 Signed-off-by: kingbri --- .github/workflows/build-wheels.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/build-wheels.yml b/.github/workflows/build-wheels.yml index d0debf176..29385165b 100644 --- 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