TensorFlow安装编译完成后,运行测试程序
#include <stdlib.h>
#include "tensorflow/core/public/session.h"
using namespace tensorflow;
using namespace std;
static Status ReadEntireFile(tensorflow::Env* env, const string& filename,
Tensor* output);
int main(int argc, char *argv[])
{
SessionOptions sessionOptions;
Session *session = NewSession(sessionOptions);
return EXIT_SUCCESS;
}
编译测试文件:
g++ tf2.cpp -o tf2 -Iport_def -Ieigen_archive -L/usr/local/include/ -I. -ltensorflow_cc -ltensorflow_framework -std=c++11 -I/usr/local/include/eigen3
/usr/local/include/是TensorFlow库文件路径
编译失败,报错:
我估计是部分库不兼容导致的
其中tensorflow和third_party需要重新从编译文件夹总复制过来,缺少的部分文件可以从/home/gosun/.cache/bazel/_bazel_gosun/13de76fce18129464b7e1e914b27c57e/execroot/org_tensorflow/bazel-out/genfiles/tensorflow文件夹总复制,但是不能替换源文件。
将eigen3也重新从安装库中复制整个文件夹下来。
之后再重新编译即可。