平台: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
文件。我希望有人能够帮助我在这里帮助我,这里有从源头建立张量流的经验。