🐛 update CMakeList

Linaom1214 · Jul 4, 2022 · fd8c9c6 · fd8c9c6
1 parent 9525cbc
commit fd8c9c6
Showing 1 changed file with 4 additions and 13 deletions.
diff --git a/yolov6/cpp/CMakeLists.txt b/yolov6/cpp/CMakeLists.txt
@@ -15,27 +15,18 @@ 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")
 
 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)