From eadf8170a676cedc6ab4f1928567a1ed3cbee050 Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期二, 06 七月 2021 10:15:06 +0800
Subject: [PATCH] up cuda
---
CMakeLists.txt | 16 ++++++----------
1 files changed, 6 insertions(+), 10 deletions(-)
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 4256d18..0682fcf 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -53,10 +53,8 @@
source_group("Source" FILES ${sources})
-#cudaS
-find_package(CUDA REQUIRED)
-
-#message("-- CUDA version:$(CUDA_VERSION)")
+link_directories(/usr/local/cuda/lib64)
+link_directories(/usr/local/lib)
# 璁剧疆鐜鐨勬ā鍨嬫煡鎵撅紝濡傛灉CUDA鐗堟湰澶т簬9,鍒欐墽琛孋UDA鐜鐨勬ā鍨嬫煡鎵�
if(CUDA_VERSION_MAJOR GREATER 9)
@@ -72,11 +70,9 @@
endif()
-set(OpenCV_DIR /data/disk2/opt/01_opencv/opencv4.5.1/build)
-FIND_PACKAGE(OpenCV REQUIRED)
-
-link_directories(/usr/local/cuda-10.0/lib64)
-#link_directories(/usr/local/lib)
+# set(OpenCV_DIR /data/disk2/opt/01_opencv/opencv4.5.1/build)
+FIND_PACKAGE(OpenCV REQUIRED PATHS "/data/disk2/opt/01_opencv/opencv4.5.2")
+include_directories( ${OpenCV_INCLUDE_DIRS} )
SET(LIBS
gomp
@@ -112,7 +108,7 @@
#include_directories("/usr/include/json/")
#include_directories("/usr/include")
#include_directories("/usr/local/include/")
-include_directories(${OPENCV_DIR}/include)
+
#璁剧疆閾炬帴搴撴悳绱㈢洰褰曪紝椤圭洰涓璴ib璺緞
link_directories(${PYTORCH_DIR}/lib) #鍔ㄦ�侀摼鎺ュ簱鐩綍/bazel-bin/tensorflow
--
Gitblit v1.8.0