Skip to content

Commit bef9d73

Browse files
gs-olivelaikhtewari
authored andcommitted
fix: Update index in installer for CI failures (#2679)
1 parent 1cf6485 commit bef9d73

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

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

+3-2
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,9 @@
22
set -eou pipefail
33
# Source conda so it's available to the script environment
44
source ${BUILD_ENV_FILE}
5-
${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision pyyaml
5+
${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision
6+
${CONDA_RUN} python -m pip install pyyaml mpmath==1.3.0
67
export TRT_VERSION=$(${CONDA_RUN} python -c "import versions; versions.tensorrt_version()")
78
${CONDA_RUN} python -m pip install /opt/torch-tensorrt-builds/torch_tensorrt*+${CU_VERSION}*.whl tensorrt~=${TRT_VERSION} tensorrt-bindings~=${TRT_VERSION} --extra-index-url=https://pypi.ngc.nvidia.com
89

9-
echo -e "Running test script";
10+
echo -e "Running test script";

0 commit comments

Comments
 (0)