# TENSORRT
set(TRT_VERSION 7.2.3.4)
set(TRT_PATH "path to tensorrt"/TensorRT-${TRT_VERSION})
# TensorRT
MESSAGE("\nTensorRT " ${TRT_VERSION})
MESSAGE(STATUS "${TRT_PATH}\n")
set(TRT_INCLUDE_PATH ${TRT_PATH}/include)
set(TRT_LIB_PATH ${TRT_PATH}/lib)
set(TRT_LIBS nvinfer nvonnxparser)
set(TRT_DLLS ${TRT_LIB_PATH}/nvinfer.dll
${TRT_LIB_PATH}/nvonnxparser.dll
${TRT_LIB_PATH}/nvinfer_plugin.dll
${TRT_LIB_PATH}/myelin64_1.dll
)
include_directories(${TRT_INCLUDE_PATH})
link_directories(${TRT_LIB_PATH})
link_libraries(${TRT_LIBS})
# do something
# ...
# ...
# when build excutable
add_custom_command(TARGET ${project}
POST_BUILD
COMMAND ${CMAKE_COMMAND} -E copy
${TRT_DLLS}
${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/${CMAKE_BUILD_TYPE}
)
etc/cmake
- [cmake] window에서 TensorRT 링킹하기 2021.08.26
[cmake] window에서 TensorRT 링킹하기
2021. 8. 26. 12:29