Temporarily install torch in test index

Author: HolyWu

.github/scripts/install-torch-tensorrt-windows.sh

@@ -1,6 +1,7 @@
 set -eou pipefail
 source "${BUILD_ENV_FILE}"
-${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision
+# ${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision
+${CONDA_RUN} pip install torch torchvision --index-url https://download.pytorch.org/whl/test/cu121
 ${CONDA_RUN} pip install pyyaml mpmath==1.3.0
 ${CONDA_RUN} pip install tensorrt==10.0.0b6 tensorrt-cu12-bindings==10.0.0b6 tensorrt-cu12-libs==10.0.0b6 --extra-index-url https://pypi.nvidia.com
 ${CONDA_RUN} pip install ${RUNNER_ARTIFACT_DIR}/torch_tensorrt*.whl

.github/workflows/build-test-windows.yml

@@ -2,9 +2,6 @@ name: Build and test Windows wheels
 
 on: [pull_request]
 
-env:
-  CHANNEL: test
-
 jobs:
   generate-matrix:
     uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main

packaging/pre_build_script_windows.sh

@@ -1,4 +1,5 @@
 python -m pip install -U numpy packaging pyyaml setuptools wheel
+python -m pip install torch==2.3.0 --index-url https://download.pytorch.org/whl/test/cu121
 python -m pip install tensorrt==10.0.0b6 tensorrt-cu12-bindings==10.0.0b6 tensorrt-cu12-libs==10.0.0b6 --extra-index-url https://pypi.nvidia.com
 
 choco install bazelisk -y