Skip to content

Commit ebbce82

Browse files
committed
Add access to test indices Torch
1 parent fd5e330 commit ebbce82

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed
Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
source "${BUILD_ENV_FILE}"
2-
${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision
2+
${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision --extra-index-url https://download.pytorch.org/whl/test/cu121 --extra-index-url https://download.pytorch.org/whl/test/cu118
33
${CONDA_RUN} pip install pyyaml mpmath==1.3.0
4-
${CONDA_RUN} pip install ${RUNNER_ARTIFACT_DIR}/torch_tensorrt*.whl tensorrt==10.0.0b6 --extra-index-url https://pypi.nvidia.com
4+
${CONDA_RUN} pip install tensorrt==10.0.0b6 --extra-index-url https://pypi.nvidia.com
5+
${CONDA_RUN} pip install ${RUNNER_ARTIFACT_DIR}/torch_tensorrt*.whl
56

67
echo -e "Running test script";

0 commit comments

Comments
 (0)