| | |
| | | set(CUDA_Path /usr/local/cuda) |
| | | # find_package(OpenCV REQUIRED PATHS "/opt/toolkits/opencv") |
| | | set(OPENCV_DIR /opt/toolkits/opencv) |
| | | set(TensorRT_DIR /opt/toolkits/TensorRT-7.2.3.4) |
| | | set(TensorRT_DIR /opt/toolkits/TensorRT7) |
| | | set(LOG_DIR /opt/toolkits/log4cplus) |
| | | |
| | | include_directories(${OPENCV_DIR}/include/opencv4) |
| | |
| | | link_directories(${LOG_DIR}/lib) |
| | | include_directories(${LOG_DIR}/include) |
| | | include_directories("/usr/include/jsoncpp/") |
| | | include_directories("/usr/include/eigen3/") |
| | | |
| | | file(GLOB_RECURSE sources src/* src/core/* src/utils/* src/detecter_tools/* ${TensorRT_DIR}/samples/common/*.h ${TensorRT_DIR}/samples/common/logger.cpp) |
| | | set(LIBS nvinfer nvparsers nvonnxparser nvinfer_plugin stdc++fs jsoncpp log4cplus opencv_core opencv_highgui opencv_imgcodecs |
| | |
| | | endif(NOT DEFINED CMAKE_CUDA_ARCHITECTURES) |
| | | |
| | | |
| | | # |
| | | find_package(CUDA REQUIRED) |
| | | if(CUDA_VERSION_MAJOR GREATER 9) |
| | | message("-- CUDA ${CUDA_VERSION_MAJOR} detected") |