File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 1
1
source " ${BUILD_ENV_FILE} "
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
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
3
3
${CONDA_RUN} pip install pyyaml mpmath==1.3.0
4
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 --extra-index-url https://download.pytorch.org/whl/test/cu121 --extra-index-url https://download.pytorch.org/whl/test/cu118
5
+ ${CONDA_RUN} pip install ${RUNNER_ARTIFACT_DIR} /torch_tensorrt* .whl torch torchvision --extra-index-url https://download.pytorch.org/whl/test/cu121 --extra-index-url https://download.pytorch.org/whl/test/cu118
6
6
7
7
echo -e " Running test script" ;
You can’t perform that action at this time.
0 commit comments