diff --git a/CMakeLists.txt b/CMakeLists.txt index 6454f1b..6750f19 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -95,3 +95,6 @@ message("install dir:" ${CMAKE_INSTALL_PREFIX}) install(DIRECTORY include/ DESTINATION include/${CMAKE_PROJECT_NAME} FILES_MATCHING PATTERN "*.h") install(TARGETS tkDNN kernels DESTINATION lib) +install(FILES "${CMAKE_SOURCE_DIR}/${CMAKE_PROJECT_NAME}Config.cmake" # source directory + DESTINATION "share/${CMAKE_PROJECT_NAME}/cmake/" # target directory +) diff --git a/tkDNNConfig.cmake b/tkDNNConfig.cmake new file mode 100644 index 0000000..8665415 --- /dev/null +++ b/tkDNNConfig.cmake @@ -0,0 +1,10 @@ +find_package(CUDA REQUIRED) +find_package(OpenCV REQUIRED) +find_library(NVINFER NAMES nvinfer) +if(NVINFER STREQUAL "NVINFER-NOTFOUND") + set(NVINFER_INCLUDES "/usr/local/nvidia/tensorrt/include/") + link_directories(/usr/local/nvidia/tensorrt/targets/x86_64-linux-gnu/lib/ + /usr/local/cuda/targets/x86_64-linux/lib/) +endif() +set(tkDNN_INCLUDE_DIRS ${CUDA_INCLUDE_DIRS} ${OPENCV_INCLUDE_DIRS} ${NVINFER_INCLUDES}) +set(tkDNN_LIBRARIES tkDNN kernels ${CUDA_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES} -lcudnn -lnvinfer ${OpenCV_LIBS})