config

2018-12-03 22:04:04 +01:00
parent a13bc2f007
commit 443179359d
2 changed files with 13 additions and 0 deletions
@@ -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})