Skip to content

Commit b8a39ac

Browse files
committed
I do need extra index for Windows 12.4 after all
1 parent b900774 commit b8a39ac

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

cibuildwheel_support/prepare_build_environment_windows_cu124.sh

+7
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,10 @@ rm cuda.exe
1010

1111
export WITH_CUDA=1
1212
export CUDA_HOME="$CUDA_ROOT"
13+
14+
# Create pip directory if it doesn't exist
15+
mkdir -p "C:\ProgramData\pip"
16+
17+
# Create pip.ini file with PyTorch CUDA 12.4 index
18+
echo "[global]
19+
extra-index-url = https://download.pytorch.org/whl/cu124" > "C:\ProgramData\pip\pip.ini"

0 commit comments

Comments
 (0)