[tensorRT] CMakelists.txt 에 tensorRT 추가

 

CMakelists.txt에 다음 내용추가

 

 

find_package(CUDA REQUIRED)
list(APPEND PLUGINS "nvinfer")
list(APPEND PLUGINS "nvonnxparser")
list(APPEND PLUGINS "nvparsers")

foreach(libName ${PLUGINS})
    find_library(${libName}_lib NAMES ${libName} "/usr" PATH_SUFFIXES lib)
    list(APPEND PLUGIN_LIBS "${${libName}_lib}")
endforeach()

include_directories(
    ${CUDA_INCLUDE_DIRS}
    ${CUDNN_ROOT_DIR}/include
)
include_directories(/usr/lib/aarch64-linux-gnu)

 

target_link_libraries(your target

${PLUGIN_LIBS}

)

 

 

bashrc 파일에 다음 내용 추가

export PATH="/usr/local/cuda-11.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH"

export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64
export PATH=$PATH:$CUDA_HOME/bin

 

 

 

 

* Reference

https://github.com/NVIDIA/TensorRT/blob/release/8.6/CMakeLists.txt

 

https://forums.developer.nvidia.com/t/cmake-cant-find-tensorrt/81019/2

 

https://forums.developer.nvidia.com/t/cudnn-linking-error-using-cmake/57968/2