Fixup

FluxML · Apr 27, 2024 · 66d3605 · 66d3605
1 parent 5ad79e0
commit 66d3605
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/BuildCWrapper.yaml b/.github/workflows/BuildCWrapper.yaml
@@ -83,6 +83,7 @@ jobs:
         arch:
           - x64
         cuda_version:
+          - "10.2.89"
           - "11.3.1"
         cudnn_version:
           - "8.2.4"
@@ -159,8 +160,7 @@ jobs:
           for F in $CUDNN_ROOT/lib/libcudnn*; do ln -sf $F /usr/local/cuda/lib64/$(basename $F); done
           cd -
           export TORCH_VARIANT="cu$(echo $CUDA_VERSION | cut -d . -f 1-2 | tr -d '.')"
-          cd /usr/local
-          wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-$TORCH_VERSION%2B$TORCH_VARIANT.zip
+          wget https://download.pytorch.org/libtorch/$TORCH_VARIANT/libtorch-cxx11-abi-shared-with-deps-$TORCH_VERSION%2B$TORCH_VARIANT.zip
           unzip -q libtorch-*.zip
           rm libtorch-*.zip
           cd -