tensorflow源码编译教程_【填坑】基于TensorFlowC++API的gRPC服务

之前实习的时候训练一个给ASR文本添加大小写和标点的模型&＃xff0c;框架用的是tensorflow r1.2(本文其实和tensorflow版本无关)。模型训好后mentor说要转成C&＃43;&＃43;上线&＃xff0c;当时差点崩溃&＃xff0c;由于太懒&＃xff0c;不想换框架重写就只好试试tensorflow的C&＃43;&＃43; API了&＃xff0c;由于公司服务器的权限问题也是躺了不少的坑&＃xff0c;这里简单总结一下TF模型转C&＃43;&＃43; API以及转gRPC服务的基本步骤和遇到的一些很迷的Errors。
关于Tensorflow模型到gRPC服务&＃xff0c;tensorflow有个神奇API叫Tensorflow Serving&＃xff0c;大家可以试一试。不过本文不是采用这种方式&＃xff0c;而是先转C&＃43;&＃43;接口&＃xff0c;再用gRPC写接口服务&＃xff0c;其实原理是一样的。

Tensorflow C&＃43;&＃43; API

Tensorflow提供的C&＃43;&＃43;API能够恢复python训练好的模型计算图和参数到C&＃43;&＃43;环境中&＃xff1b;通过向Placeholder传入数据便可以得到Eigen::Tensor类型的返回。python下的Tensorflow依赖numpy矩阵运算库&＃xff0c;而在C&＃43;&＃43;下依赖Eigen::Tensor库&＃xff0c;所以在使用C&＃43;&＃43; API之前需要先安装好对应版本的Eigen库&＃xff1b;
因为模型最后是一层CRF&＃xff0c;所以还需要用Eigen重写Viterbi解码&＃xff0c;还好只是简单的DP问题&＃xff1b;这里简单介绍一下提到的模型的结构&＃xff1a;525通道CNN &＃43; HighWayNet &＃43; bi-LSTM &＃43; CRF&＃xff1b;

下载Tensorflow源码

下载最新的Tensorflow源码&＃xff0c;这个和你使用什么版本Tensorflow训练模型没有关系。之后就需要把Tensorflow编译成我们需要的动态链接库&＃xff1b;

安装Bazel

这里需要注意一下&＃xff0c;版本太新和太旧的Bazel在编译Tensorflow的时候都会报错&＃xff0c;这里举例我用过的版本组合&＃xff1a;Bazel-0.10.0(Tensorflow-r1.7)&＃xff1b;Bazel-0.8.0(Tensorflow-r1.5)&＃xff1b;Bazel-0.4.5(Tensorflow-r1.2)。以上组合并不固定&＃xff0c;经供参考(本文是Tensorflow1.7)。由于在公司服务器上工作&＃xff0c;所有的third-party都需要安装在自己的目录。
1 . 非root安装JDK8&＃xff1a;jdk-8u161-linux-x64.tar.gz。Bazel依赖JDK8&＃xff0c;wget下载后解压&＃xff0c;把jdk添加到环境变量&＃xff0c;把以下代码添加到$HOME/.bashrc&＃xff1a;

2 . 安装Bazel&＃xff1a;各版本地址release&＃xff0c;本文是bazel-0.10.0&＃xff0c;下载好.sh文件之后执行一下命令:

bazel被装到了$HOME/bin目录下&＃xff0c;添加到环境就OK了&＃xff1b;之后输入bazel version看看版本是否安装成功&＃xff1b;

安装Eigen3

之前提到了Tensorflow依赖Eigen矩阵运算库&＃xff0c;在编译之前需要安装对应的版本&＃xff1b;关于Eigen同样是一个坑&＃xff0c;不对应的版本依然会让Tensorflow编译失败&＃xff0c;这里提供一个最保险的方法&＃xff0c;就是去tensorflow的tensorflow/tensorflow/workspace.bzl里下载&＃xff1b;在 workspace.bzl中找到&＃xff1a;

下载其中任何一个链接都可&＃xff0c;下载好之后解压&＃xff0c;将Eigen添加到环境变量&＃xff1a;

安装Protobuf

protocbuf是一种很强大的跨平台的数据标准&＃xff0c;可以用于结构化数据序列化&＃xff0c;用于通讯协议、数据存储等领域的语言无关、平台无关的序列化结构数据格式&＃xff0c;在之后的gRPC中也会用到&＃xff1b;
同样Protobuf的版本也会直接决定tensorflow是否编译成功&＃xff0c;和安装Eigen同样的方法&＃xff0c;去workspace.bzl中找protobuf下载对应的版本&＃xff1b;下载好后进入protobuf目录输入以下命令安装&＃xff0c;并添加到环境&＃xff1a;

安装nsync

和Eigen同样的方式下载&＃xff0c;添加环境路径即可&＃xff1a;

跳过这步安装会出现&＃xff1a;fatal error : nsync_cv.h: No such file or dictionary的错误&＃xff1b;

编译Tensorflow

经过以上的充足准备&＃xff0c;终于可以编译Tensorflow啦&＃xff0c;进入tensorflow下载目录&＃xff0c;输入以下命令&＃xff1a;

其中./configure之后没有选择CUDA支持&＃xff0c;全部为no。经过4/5分钟之后&＃xff0c;在bazel-bin/tensorflow下就会看到libtensorflow_cc.so和libtensorflow_framework.so两个动态库&＃xff1b;之后需要把这两个库复制到$HOME/tools/lib中&＃xff0c;这样就可以连接来编译我们的模型了&＃xff0c;之后的任务就是写Tensorflow的C&＃43;&＃43; API接口啦。

C&＃43;&＃43;重写python API

在python API中主要有以下三步骤&＃xff1a;
1 . 创建Session&＃xff0c;读入计算图&＃xff0c;恢复参数&＃xff1b;
2 . 获取需要的输入&＃xff0c;输出Tensor (graph.get_tensor_by_name);
3 . 给输入Tensor传值&＃xff0c;run模型&＃xff0c;得到输出结果&＃xff1b;
C&＃43;&＃43; API也是相同的步骤&＃xff1b;这里先给出python API的代码&＃xff0c;Tensor的名字最好提前设定好&＃xff0c;如果没有的话也可以直接Tensor.name查看&＃xff1a;

同样的结构用C&＃43;&＃43;重写之后的代码如下&＃xff0c;恢复模型部分&＃xff1a;

API核心函数&＃xff0c;C&＃43;&＃43;中Tensor返回的是Eigen::Tensor类型&＃xff1b;

如果模型输出不是最终结果&＃xff0c;还需要进行行加工&＃xff0c;这时就需要对Eigen的API有稍微的了解了&＃xff0c;我用Eigen写了一个简单的CRF-Viterbi_decode代码&＃xff0c;分享在这里供大家参考&＃xff1a;

编译TF模型

通过以上步骤&＃xff0c;我们就可以编译C&＃43;&＃43; API了&＃xff0c;这里我们用make进行编译&＃xff0c;链接上之前编译的libtensorflow_cc.so和libtensorflow_framework.so&＃xff0c;命令如下&＃xff0c;也可以写一个Makefile&＃xff1b;

之后在当前目录会生成一个可执行文件&＃xff0c;这样就大功告成啦&＃xff5e;输入一句没有标点的句子试一试&＃xff0c;得到如下结果&＃xff0c;试验成功&＃xff1b;

Tensorflow gRPC服务

有了C&＃43;&＃43; API就可以愉快的写gRPC服务了&＃xff0c;那gPRC服务到底是什么呢&＃xff1f;google家的RPC&＃xff0c;传送官方文档&＃xff1a;定义一个服务&＃xff0c;指定其能够被远程调用的方法&＃xff08;包含参数和返回类型&＃xff09;。客户端应用可以像调用本地对象一样直接调用另一台不同的机器上服务端应用的方法。

• 安装gRPC
  直接github clone就好了&＃xff1a;

进入grpc目录&＃xff0c;更新三方依赖源码&＃xff0c;由于我们安装了protobuf&＃xff0c;所以可以进入.gitmodules文件&＃xff0c;删掉protobuf项&＃xff0c;之后将已经安装的protobuf目录放入grpc/third_party就好&＃xff1a;

更新完之后进入Makefile查找&＃xff1a;ldconfig&＃xff0c;把动态链接库指向自己的目录&＃xff1b;

把ldconfig替换成ldconfig -r $HOME/tools/bin&＃xff0c;之后执行安装&＃xff1a;

如果不是在grpc/third_party中安装的protobuf&＃xff0c;在make过程中很有可能出如下错误&＃xff1a;

可以通过如下方式来解决这个错误&＃xff1a;

最新版本的grpc需要protobuf的版本是3.5.0&＃xff0c;安装成功之后可以去/grpc/examples/cpp/下测试grpc是否能正常工作&＃xff1b;如果安装的protobuf版本不对会报错&＃xff0c;更新protobuf到3.5.0即可&＃xff0c;注意还要和Tensorflow要求的protobuf版本匹配才行&＃xff1b;

Tensorflow C&＃43;&＃43; API 的gRPC服务

到这里总算可以开始写服务了&＃xff0c;在实际运用中要求服务的client和server端都能够异步工作&＃xff0c;也就是请求不产生阻塞&＃xff1b;gPRC提供了很强的异步服务机制来实现客户和服务之间的异步无阻塞&＃xff0c;这里将简单分析一下client和server端的异步机制&＃xff1a;

1 . Client端&＃xff1a;客户端会生成一个队列CompletionQueue&＃xff0c;并用CallData类来记录RPC的状态和标签&＃xff0c;每个request对应一个Calldata&＃xff0c;在接收到请求的时候将其放入CompletionQueue中&＃xff0c;并调用Finish函数向服务器端发送请求&＃xff0c;寻求应答后立即返回处理新的待发送请求(无阻塞)&＃xff1b;另开一个线程去等待处理CompletionQueue中的服务端应答&＃xff1b;

2 . Server端&＃xff1a;服务端有两个任务&＃xff1a;接收request和处理request并返回Client&＃xff1b;服务端用ServerData类来接收request&＃xff0c;为了不让Service处理请求过程中有新的request到来产生阻塞&＃xff0c;服务端将ServerData放入CompletionQueue队列后新建一个ServerData去接收新的请求(无阻塞)&＃xff1b;另开一个线程处理队列中各种状态的ServiceData&＃xff0c;并实现应答&＃xff1b;

以上是我自己的理解&＃xff0c;如果有错误请大家指出&＃xff1b;根据这种理解实现Tensorflow的gRPC服务就不难了&＃xff0c;服务端在创建Service之前先restore model&＃xff0c;服务启动之后直接调用API即可&＃xff0c;异步的实现和上面提到的流程一样&＃xff0c;grpc有一个官方的案例非常不错/grpc/examples/cpp/helloworld/helloworld_async_client2.cc。
测试Tensorflow gRPC的结果如下&＃xff1a;

总结

这么折腾下来总算是完成了Mentor的任务了&＃xff0c;感觉大部分的时间都花在安装三方库和配环境上&＃xff0c;不过也是有收获的&＃xff0c;这篇文章作为这一套工作的简单总结&＃xff0c;文章中的错误或者过时的东西请各位看官大神们大声说出来呀&＃xff5e;&＃xff5e;时间不早了&＃xff0c;明儿还要实习&＃xff0c;晚安&＃xff5e;