We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents cd86660 + 76c9ebd commit 3ece71bCopy full SHA for 3ece71b
.github/scripts/install-torch-tensorrt.sh
@@ -2,9 +2,9 @@
2
set -eou pipefail
3
# Source conda so it's available to the script environment
4
source ${BUILD_ENV_FILE}
5
-${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision --extra-index-url https://pypi.python.org/simple
+${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision
6
${CONDA_RUN} python -m pip install pyyaml mpmath==1.3.0
7
export TRT_VERSION=$(${CONDA_RUN} python -c "import versions; versions.tensorrt_version()")
8
${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
9
10
-echo -e "Running test script";
+echo -e "Running test script";
0 commit comments