From 3db1be412cee72a954ac83144d8512f5cba4b139 Mon Sep 17 00:00:00 2001 From: Morgan Funtowicz Date: Tue, 23 Jul 2024 07:27:34 +0000 Subject: [PATCH] commenting out Python part for TensorRT installation --- backends/trtllm/scripts/install_tensorrt.sh | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/backends/trtllm/scripts/install_tensorrt.sh b/backends/trtllm/scripts/install_tensorrt.sh index 65111ef5..4472bf9a 100755 --- a/backends/trtllm/scripts/install_tensorrt.sh +++ b/backends/trtllm/scripts/install_tensorrt.sh @@ -75,8 +75,8 @@ install_centos_requirements() { } install_tensorrt() { - PY_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[0:2])))') - PARSED_PY_VERSION=$(echo "${PY_VERSION//./}") + #PY_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[0:2])))') + #PARSED_PY_VERSION=$(echo "${PY_VERSION//./}") TRT_CUDA_VERSION="12.4" if [ -z "$RELEASE_URL_TRT" ];then @@ -91,7 +91,7 @@ install_tensorrt() { wget --no-verbose ${RELEASE_URL_TRT} -O /tmp/TensorRT.tar tar -xf /tmp/TensorRT.tar -C /usr/local/ mv /usr/local/TensorRT-${TRT_VER} /usr/local/tensorrt - pip3 install /usr/local/tensorrt/python/tensorrt-*-cp${PARSED_PY_VERSION}-*.whl + # pip3 install /usr/local/tensorrt/python/tensorrt-*-cp${PARSED_PY_VERSION}-*.whl rm -rf /tmp/TensorRT.tar } @@ -114,4 +114,4 @@ case "$ID" in echo "Unable to determine OS..." exit 1 ;; -esac \ No newline at end of file +esac