Tensorflow C++ 学习（一） 搭建环境

转自：http://blog.csdn.net/jmh1996/article/details/73197337?locationNum=6

前言

Tensorflow 网上大部分是python的资料较多，而C++方面的极少，因此，接下来会有一系列的博客用于学习tensorflow,记录学习的过程。加油！

搭建环境

既然使用C++的API，那第一步就是搭建Tensorflow的工作环境 

1. 准备一台64位的虚拟机 ，我安装的ubuntu 16.04 64位的. 用新的虚拟机主要是图个干净利落，同时修改好软件源，建议改成阿里云的，安装一些常用的运行库，开发库。 

2. 安装 bazel
> echo
"deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8"
| sudo tee /etc/apt/sources.list.d/bazel.listcurl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add
->sudo apt-get update && sudo apt-get install bazel

可能会提示没有安装curl，则需要先安装curl

> sudo apt-get install curl

1

安装bazel

>sudo apt-get update && sudo apt-get install bazel

1

本人使用 ubuntu 16.04刚上的虚拟机，妥妥的。 

3. 安装python的一些常用库

>sudo apt-get install sudo apt-get install python-numpy python-dev python-pip python-wheel -y

1
安装cpu 开发工具库

>sudo apt-get install libcupti-dev

1
下载tensorflow 源码

>sudo apt-get install git
>git clone https://github.com/tensorflow/tensorflow >cd tensorflow
>./configure #这一步不可少！否则后面c++编译不过

1
2
3
4
5

出现这个就配置对了：

INFO:All external dependencies fetched successfully.
安装tensorflow ，使用pip

>pip install --upgrade tensorflow

1
测试C++接口

将 tensorflow目录下的：tensorflow/cc/example/example.cc 文件内容替换为（没有就新建一个这个目录下的文件）:

// tensorflow/cc/example/example.cc

#include "tensorflow/cc/client/client_session.h"
#include "tensorflow/cc/ops/standard_ops.h"
#include "tensorflow/core/framework/tensor.h"

int main() {
using namespace tensorflow;
using namespace tensorflow::ops;
Scope root = Scope::NewRootScope();
// Matrix A = [3 2; -1 0]
auto A = Const(root, { {3.f, 2.f}, {-1.f, 0.f}});
// Vector b = [3 5]
auto b = Const(root, { {3.f, 5.f}});
// v = Ab^T
auto v = MatMul(root.WithOpName("v"), A, b, MatMul::TransposeB(true));
std::vector<Tensor> outputs;
ClientSession session(root);
// Run and fetch v
TF_CHECK_OK(session.Run({v}, &outputs));
// Expect outputs[0] == [19; -3]
std::cout<< outputs[0].matrix<float>();
return 0;
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

在tensorflow/cc/example/ 目录下新建一个BUILD文件，文件内容：

cc_binary(
name = "example",
srcs = ["example.cc"],
deps = [
"//tensorflow/cc:cc_ops",
"//tensorflow/cc:client_session",
"//tensorflow/core:tensorflow",
],
)

1
2
3
4
5
6
7
8
9

编译：

>cd tensorflow #回到tensorflow的主目录下
>bazel run -c opt //tensorflow/cc/example:example

1
2

然后首次编译很久，等一会儿就好。

最后打印出：

2017-06-13 17:20:50.578854: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations.
2017-06-13 17:20:50.578970: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations.
2017-06-13 17:20:50.578992: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX instructions, but these are available on your machine and could speed up CPU computations.
19
-3

1
2
3
4
5

说明安装成功。 

但是！这里有太多的调试信息了，我们把它去掉：

>export TF_CPP_MIN_LOG_LEVEL=2
>bazel-bin/tensorflow/cc/example/example

1
2

打印：

19
-3

1
2

OK Perfect!