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