Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 3751ac9

Browse files
committedMar 9, 2024
fix: Updatempmath version
1 parent 758f820 commit 3751ac9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
 

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
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 --extra-index-url https://pypi.python.org/simple
5+
${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision pyyaml mpmath==1.3.0 --extra-index-url https://pypi.python.org/simple
66
export TRT_VERSION=$(${CONDA_RUN} python -c "import versions; versions.tensorrt_version()")
77
${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
88

0 commit comments

Comments
 (0)
Please sign in to comment.