From fd8c9c666dbb31abc2af365bbd22cea0a5b0aae6 Mon Sep 17 00:00:00 2001 From: linaom1214 Date: Mon, 4 Jul 2022 18:15:42 +0800 Subject: [PATCH] :bug: update CMakeList --- yolov6/cpp/CMakeLists.txt | 17 ++++------------- 1 file changed, 4 insertions(+), 13 deletions(-) diff --git a/yolov6/cpp/CMakeLists.txt b/yolov6/cpp/CMakeLists.txt index ba9f31a..bae6f0d 100644 --- a/yolov6/cpp/CMakeLists.txt +++ b/yolov6/cpp/CMakeLists.txt @@ -15,15 +15,9 @@ include_directories(${PROJECT_SOURCE_DIR}/include) # cuda include_directories(/usr/local/cuda/include) link_directories(/usr/local/cuda/lib64) -# cudnn -# include_directories(/data/cuda/cuda-10.2/cudnn/v8.0.4/include) -# link_directories(/data/cuda/cuda-10.2/cudnn/v8.0.4/lib64) - - - # tensorrt -include_directories(/home/mat/anaconda3/envs/pytorch/lib/python3.8/site-packages/tensorrt/include) -link_directories(/home/mat/anaconda3/envs/pytorch/lib/python3.8/site-packages/tensorrt) +include_directories(/usr/include/x86_64-linux-gnu/) +link_directories(/usr/lib/x86_64-linux-gnu/) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Ofast -Wfatal-errors -D_MWAITXINTRIN_H_INCLUDED") @@ -31,11 +25,8 @@ find_package(OpenCV) include_directories(${OpenCV_INCLUDE_DIRS}) add_executable(yolov6 ${PROJECT_SOURCE_DIR}/yolov6.cpp) -target_link_libraries(yolov6 /home/mat/anaconda3/envs/pytorch/lib/python3.8/site-packages/tensorrt/libnvinfer.so.8) +target_link_libraries(yolov6 nvinfer) target_link_libraries(yolov6 cudart) target_link_libraries(yolov6 ${OpenCV_LIBS}) -add_definitions(-O2 -pthread) - - - +add_definitions(-O2 -pthread) \ No newline at end of file