使用Tensorflow的C ++ API构建错误

Question

I'm building the following code with cmake . I want to use tensorflow's C++ API,but I encountered errors when compiling the code.

Here are the errors:

  >/home/zifeng/software/clion-2017.2.3/bin/cmake/bin/cmake --build  
  >/home/zifeng/CLionProjects/tfcpp_demo/cmake-build-debug --target all -- -j 4  
  >[ 50%] Building CXX object CMakeFiles/tfcpp_demo.dir/src/main.cpp.o  
  >[100%] Linking CXX executable tfcpp_demo  
  >/usr/bin/ld: CMakeFiles/tfcpp_demo.dir/src/main.cpp.o: undefined reference to symbol '_ZNK10tensorflow6Status8ToStringB5cxx11Ev'  
  >//home/zifeng/CLionProjects/tfcpp_demo/./lib/libtensorflow_framework.so: error adding symbols: DSO missing from command line  
  >collect2: error: ld returned 1 exit status  
  >CMakeFiles/tfcpp_demo.dir/build.make:94: recipe for target 'tfcpp_demo' failed  
  >make[2]: *** [tfcpp_demo] Error 1  
  >CMakeFiles/Makefile2:67: recipe for target 'CMakeFiles/tfcpp_demo.dir/all' failed  
  >make[1]: *** [CMakeFiles/tfcpp_demo.dir/all] Error 2  
  >Makefile:83: recipe for target 'all' failed  
  >make: *** [all] Error 2 

Here is my code:

#include <tensorflow/core/platform/env.h>
#include <tensorflow/core/public/session.h>

#include <iostream>

using namespace std;
using namespace tensorflow;

int main()
{
    Session* session;
    Status status = NewSession(SessionOptions(), &session);
    if (!status.ok()) {
        cout << status.ToString() << "\n";
        return 1;
    }
    cout << "Session successfully created.\n";
}

Here is my makefile:

cmake_minimum_required(VERSION 3.7)
project(tfcpp_demo)

set(CMAKE_CXX_STANDARD 11)
link_directories(./lib)

set(SOURCE_FILES
        src/main.cpp)

include_directories(
        /home/zifeng/software/tensorflow/tensorflow
        /home/zifeng/software/tensorflow/tensorflow/bazel-genfiles
        /home/zifeng/software/tensorflow/tensorflow/tensorflow/contrib/makefile/gen/protobuf/include
        /home/zifeng/software/tensorflow/tensorflow/tensorflow/contrib/makefile/downloads/nsync/public
        /usr/local/lib/python2.7/dist-packages/tensorflow/include
)
add_executable(tfcpp_demo  ${SOURCE_FILES})
target_link_libraries(tfcpp_demo tensorflow_cc)

Answer 1

I have same problem with you and I fix it.

It looks some libtensorflow .so files are missing at linking.

I generate my libtensorflow_cc.so by using two command:

bazel build --config=monolithic --config=cuda //tensorflow:libtensorflow_cc.so
bazel build --config=monolithic --config=cuda //tensorflow:libtensorflow_framework.so

It is better than one command.

bazel build --config=opt --config=cuda //tensorflow:libtensorflow_cc.so

Answer 2

#include <tensorflow/core/platform/env.h>
#include <tensorflow/core/public/session.h>

#include <iostream>

using namespace std;
using namespace tensorflow;

int main()
{
    Session* session;
    Status status = NewSession(SessionOptions(), &session);
    if (!status.ok()) {
        cout << status.ToString() << "\n";
        return 1;
    }
    cout << "Session successfully created.\n";
}