# 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}
)

+ Recent posts