Improve caffe detection function.

2017-10-09 14:42:30 +02:00
parent 87da22390b
commit 96b3b695ab
2 changed files with 41 additions and 13 deletions
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -30,8 +30,8 @@ target_link_libraries(fmri ${Boost_LIBRARIES})

 # Require Caffe
 find_package(Caffe REQUIRED)
-include_directories(${Caffe_INCLUDE_DIR})
-target_link_libraries(fmri ${Caffe_LIBS})
+include_directories(${Caffe_INCLUDE_DIRS})
+target_link_libraries(fmri ${Caffe_LIBRARIES})

 # Require glog
 find_package(Glog REQUIRED)