File tree 1 file changed +3
-2
lines changed
1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -24,14 +24,15 @@ check_library_exists("libtensorflow.so" "" "" tensorflow_systemwide)
24
24
set (INSTALL_HEADERS "src/Tensor.h;src/Model.h" )
25
25
26
26
if (tensorflow_systemwide)
27
+ message (STATUS "Looking for tensorflow systemwide" )
27
28
set (LIBS tensorflow)
28
29
get_property (dirs DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY INCLUDE_DIRECTORIES )
29
30
foreach (dir ${dirs} )
30
31
message (STATUS "dir='${dir} '" )
31
32
include_directories (${CMAKE_PROJECT_NAME} PRIVATE SYSTEM dirs/tensorflow/c)
32
33
endforeach ()
33
-
34
34
else ()
35
+ message (STATUS "Looking for tensorflow in ${CMAKE_SOURCE_DIR} /libtensorflow/lib/" )
35
36
set (LIBS ${CMAKE_SOURCE_DIR} /libtensorflow/lib/libtensorflow.so)
36
37
include_directories (${CMAKE_PROJECT_NAME} PRIVATE libtensorflow/include /tensorflow/c)
37
38
set (INSTALL_HEADERS "${INSTALL_HEADERS} ;libtensorflow/include/tensorflow/c/c_api.h" )
@@ -53,4 +54,4 @@ install(TARGETS ${PROJECT_NAME}
53
54
54
55
add_subdirectory (examples/coco)
55
56
add_subdirectory (examples/load_model)
56
- add_subdirectory (examples/mnist)
57
+ add_subdirectory (examples/mnist)
You can’t perform that action at this time.
0 commit comments