Skip to content

Commit 9baf073

Browse files
committed
Switch to TRT 10 in installer
1 parent 090b9b9 commit 9baf073

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
source "${BUILD_ENV_FILE}"
22
${CONDA_RUN} ${PIP_INSTALL_TORCH} torchvision
33
${CONDA_RUN} pip install pyyaml mpmath==1.3.0
4-
${CONDA_RUN} pip install ${RUNNER_ARTIFACT_DIR}/torch_tensorrt*.whl tensorrt==9.3.0.post12.dev1 tensorrt-bindings==9.3.0.post12.dev1 tensorrt-libs==9.3.0.post12.dev1 --extra-index-url https://pypi.nvidia.com
4+
${CONDA_RUN} pip install ${RUNNER_ARTIFACT_DIR}/torch_tensorrt*.whl tensorrt==10.0.0b6 --extra-index-url https://pypi.nvidia.com --force-reinstall --no-cache
55

66
echo -e "Running test script";

0 commit comments

Comments
 (0)