undo changes. issue is in pytorch package

Author: stefdoerr

.github/workflows/publish.yml

@@ -62,18 +62,6 @@ jobs:
           CPU_TRAIN: ${{ runner.os == 'macOS' && 'true' || 'false' }}
           WITH_CUDA: ${{ matrix.cuda != 'cpu' && '1' || '0' }}
 
-      - name: Set CUDA arch list for Windows
-        if: matrix.os == 'windows-2019'
-        shell: bash -l {0}
-        run: |
-          if [[ "${{ matrix.cuda }}" == "cu118" ]]; then
-            echo 'TORCH_CUDA_ARCH_LIST="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9+PTX"' >> $GITHUB_ENV
-            echo 'CUDA_HOME="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8"' >> $GITHUB_ENV
-          elif [[ "${{ matrix.cuda }}" == "cu124" ]]; then
-            echo 'TORCH_CUDA_ARCH_LIST="5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0+PTX"' >> $GITHUB_ENV
-            echo 'CUDA_HOME="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.4"' >> $GITHUB_ENV
-          fi
-
       - name: Build wheels
         if: matrix.os == 'windows-2019'
         shell: cmd # Use cmd on Windows to avoid bash environment taking priority over MSVC variables
@@ -84,8 +72,6 @@ jobs:
           DISTUTILS_USE_SDK: "1" # Windows requires this to use vc for building
           SKIP_TORCH_COMPILE: "true"
           CIBW_BUILD_VERBOSITY_WINDOWS: "1"
-          TORCH_CUDA_ARCH_LIST: ${{ env.TORCH_CUDA_ARCH_LIST }}
-          CUDA_HOME: ${{ env.CUDA_HOME }}
 
       - uses: actions/upload-artifact@v4
         with:

cibuildwheel_support/prepare_build_environment_windows_cu118.sh

@@ -8,6 +8,9 @@ curl --netrc-optional -L -nv -o cuda.exe https://developer.download.nvidia.com/c
 ./cuda.exe -s nvcc_11.8 cudart_11.8 cublas_dev_11.8 curand_dev_11.8 cusparse_dev_11.8 cusolver_dev_11.8 thrust_11.8
 rm cuda.exe
 
+export WITH_CUDA=1
+export CUDA_HOME="$CUDA_ROOT"
+
 # Create pip directory if it doesn't exist
 mkdir -p "C:\ProgramData\pip"
 

cibuildwheel_support/prepare_build_environment_windows_cu124.sh

@@ -7,3 +7,6 @@ CUDA_ROOT="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.4"
 curl --netrc-optional -L -nv -o cuda.exe https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_551.61_windows.exe
 ./cuda.exe -s nvcc_12.4 cudart_12.4 cublas_dev_12.4 curand_dev_12.4 cusparse_dev_12.4 cusolver_dev_12.4
 rm cuda.exe
+
+export WITH_CUDA=1
+export CUDA_HOME="$CUDA_ROOT"