correctly export envvars for Windows

Author: stefdoerr

.github/workflows/publish.yml

@@ -68,16 +68,12 @@ jobs:
         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: just print the envvar
-        if: matrix.os == 'windows-2019'
-        run: echo $TORCH_CUDA_ARCH_LIST
-        env:
-          TORCH_CUDA_ARCH_LIST: ${{ env.TORCH_CUDA_ARCH_LIST }}
-
       - name: Build wheels
         if: matrix.os == 'windows-2019'
         shell: cmd # Use cmd on Windows to avoid bash environment taking priority over MSVC variables
@@ -89,6 +85,7 @@ jobs:
           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,9 +8,6 @@ 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,6 +7,3 @@ 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"