tensorflow c++ api 预测python训练好的模型

一.安装说明:

1.安装bazel, Eigen

1)安装bazel

本人采用的源码安装方式,在https://github.com/bazelbuild/bazel/releases上下载

与硬件配置相关的安装源码。如在linux的64位系统下,可以下载bazel-0.19.2 installer linux-x86_64.sh(注意,tensorflow 的编译对于bazel 对应版本有要求,1.13对应的是bazel-0.19.2,如果版本有误,会编译不通过),并执行shell文件安装。具体版本需求请参照

https://www.tensorflow.org/install/source

2)安装Eigen

说明: Eigen 是一个高层次的C++库,有效支持线性代数,矩阵和矢量运算,数值分析及其相关的算法。Eigen是一个开源库,从3.1.1版本开始遵从MPL2许可,如果在tensorflow的build前没有安装eigen,在bazel的build过程中会报如下错误:

/tensorflow/core/framework/tensor. types h:19:10:fatal

error: /third_ party/eigen3/unsupported/Eigen/CXX1/Tensor. No such file or directory

include "third party/eigen3/unsupported/Eigen/CXX11/Tensor"

compilation terminated.

Target //tensorflow/core::tensorfiow failed to build

安装说明:

1. apt-get方式(假设默认安装到usr/local/include里(可在终端中输入locate eigen3查看  位置),若实际中默认安装到/usrs/include的话,可以对应替换下面命令的相应部分)

运行命令:

sudo apt-get install libeigen3-dev

运行命令: (参考第二种方法的最下面的说明)

sudo cp -r /usr/local/include/eigen3/Eigein /usr/local/include执行复制命令,将Eigen文件夹及其内容放在/usr/include

2.下载源码包和安装protocbuf

git clone -recursive https://github.com/tensorflow/tensorflow

cd tensorflow/contrib/makefile

./build_all_linux.sh

./configure

至此,tensorflow 配置完毕.

 说明:安装protobuf  

Protocol Buffers  (简称Protobuf) 是Google开源的一款跨语言,跨平台,拓展性好的  序列化工具,网上有许多安装  protoc的教程,但是我推荐最好使用tesorflow源码包里面的build_ alllinux.sh进行protobuf的自动安装,这样,protoc 的版本和tensorflow的版本是对应的(import)

cd tensoflow/tensorflow/contrib/makefile && /build alllindex.sh

查看版本以验证安装: protoc --version


二.编译过程

编译过程有两种方式,种是使用 bazel build的方式,直接在源码的workspace里面去编译:另一种, 是使用先编译动态链接库。再使用编译好的so文件,实现具体的compilation任务。

1)

bazel build -C opt <target>

其中copt的含义为--compilation_ mode=opt, 也可以用以下形式表现:

config. setting(

name=" simple',

values = {"compilation. mode":"opt"})

若需要在GPU上执行,则添加一个config选项

bazel build C opt -copt=mavx -config=cuda <target>

举例说明:

编写示例程序demo.cc

#include <iostream>

#include "tensorflow/cc/client/client_session.h"

#include tensorflow/cc/ops/standard_ops.h"

int main(){

using namespace tensorflow;

using namespace tensorflow ops;

Scope root = Scope NewRootScope():

//MatrixA=[3:2:-10]

auto A= Const(root, {3.f,2.f},{-1f,0.f)\});

//Vector b=[3 5]

auto b= Const(root, {3.f,5.f });

auto v=MatMul(root WithOpName("v", A, b, MatMul:TransposeB(true));

std:vector<Tensor> outputs;

ClientSession session(root);

mkdir tensorflow/demo

demo.cc放到tensorflow/demo 文件夹

BUILD文件写为

cc_binary(

    name="demo", srcs=["demo.cc"],

    deps=["//tensorflow/cc:cc_ops",

                "//tensorflow/cc:client_session",

                "//tensorflow/core:tensorflow"]])

运行 bazel build- opt demo  生成demo可执行文件,生成可执行文件的位置在 bazel-bin/tensorflow/demo/demo

运行示例:

 bazel-bin/tensorflow/demo/demo

2019-09-0615:10:59.905602: tensorflow/demo/democc:219

2)方法二:先编译动态链接库

 bazel build//tensorflow:libtensorflow_cc.so

建立 TensorFlow库文件夹: sudo mkdir/usrr/local/tensorflow

复制 include文件:

 mkdir /usr/local/tensorflow/include

 cp -r bazel-genfiles/ /usr/local/tensorflow/include/tf  cp-r tensorflow/*/usr/local/tensorflow/include/tf/tensorflow

 cp -r third_party /usr/local/tensorflow/include/tf

复制lib文件:

 sudo mkdir /usr/local/tensorflow/lib  

 sudo cp-r bazel-bin/tensorflow/libtensorflow_cc.so/usr/local/tensorflow/lib/

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,294评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,493评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,790评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,595评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,718评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,906评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,053评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,797评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,250评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,570评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,711评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,388评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,018评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,796评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,023评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,461评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,595评论 2 350