diff --git a/FastDeploy.cmake.in b/FastDeploy.cmake.in index 3ad85fb79..44e7eb8a6 100644 --- a/FastDeploy.cmake.in +++ b/FastDeploy.cmake.in @@ -145,7 +145,7 @@ if(WITH_GPU) find_library(TRT_ONNX_LIB nvonnxparser /usr/lib/aarch64-linux-gnu/) find_library(TRT_PLUGIN_LIB nvinfer_plugin /usr/lib/aarch64-linux-gnu/) else() - if(TRT_DIRECTORY) + if(EXISTS ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/tensorrt/) find_library(TRT_INFER_LIB nvinfer ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/tensorrt/lib NO_DEFAULT_PATH) find_library(TRT_ONNX_LIB nvonnxparser ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/tensorrt/lib NO_DEFAULT_PATH) find_library(TRT_PLUGIN_LIB nvinfer_plugin ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/tensorrt/lib NO_DEFAULT_PATH) @@ -288,7 +288,6 @@ message(STATUS " ENABLE_VISION : ${ENABLE_VISION}") message(STATUS " ENABLE_TEXT : ${ENABLE_TEXT}") if(WITH_GPU) message(STATUS " CUDA_DIRECTORY : ${CUDA_DIRECTORY}") - message(STATUS " TRT_DIRECTORY : ${TRT_DIRECTORY}") endif() if(OPENCV_DIRECTORY) message(STATUS " OPENCV_DIRECTORY : ${OPENCV_DIRECTORY}")