torch version string fix

teddykoker · Jul 26, 2023 · ec52598 · ec52598
1 parent 6bf7968
commit ec52598
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -55,7 +55,7 @@ jobs:
         run: |
           export FORCE_CUDA=1
           export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
-          TORCH_VERSION=`echo "pt${{ matrix.torch-version }}" | sed "s/^\([0-9]\+\.[0-9]\+\).*$/\1/" | sed "s/\.//g"`
+          TORCH_VERSION=`echo "pt${{ matrix.torch-version }}" | sed "s/..$//" | sed "s/\.//g"`
           CUDA_VERSION=`echo ${{ matrix.cuda-version }}`
           export TORCHSORT_VERSION_SUFFIX="+$TORCH_VERSION$CUDA_VERSION"
           python3 -m pip install wheel