使用Bazel从源构建Tensorflow,无法找到op_def.pb.h

时间:2017-10-23 17:32:31

标签: c++ tensorflow

平台:Linux Ubuntu 16.04

Tensorflow:1.4.0,从源安装

Python版本:2.7

Bazel版本:0.6.1

GPU型号和内存:笔记本电脑CPU

重现的确切命令:bazel build -c opt //tensorflow/core/user_ops:rest_api.so --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"

/tensorflow/core/user_ops中的我的构建文件:

load("//tensorflow:tensorflow.bzl", "tf_custom_op_library")

cc_library(
    name = "rest_api_dependencies",
    srcs = glob(["*.c", "*.cpp"]),
    hdrs = glob(["*.h", "*.hpp", "dict.c"]),
    linkopts = ["-pthread", "-lev", "-fexceptions"],
    deps = ["//tensorflow/core:framework", 
            "//tensorflow/core:reader_base"],
)

tf_custom_op_library(
    name = "rest_api.so",
    srcs = ["redis_reader.cc"],
    deps = ["rest_api_dependencies"],
)

我一直在尝试通过bazel为我的自定义操作阅读器构建共享库,但是编译器无法找到op_def.pb.h文件。它似乎不在tensorflow/core/framework文件夹中,但它在bazel-genfiles/tensorflow/core/framework中(从之前不相关的版本生成)。所以我在源代码(rest_api.h)中向此目录添加了一个include指令,但它仍然无法找到op_def.pb.h文件。我希望有人能够帮助我在这里帮助我,这里有从源头建立张量流的经验。

0 个答案:

没有答案