1.安装bazel,根据官方教程进行安装
- 注意版本号
- 注意安装的方式,这里有两种安装方式,第一种安装方法一般是安装最新版本,第二种则是手动下载所需的版本进行安装bazel version(测试安装完成!)
2.Tensorflow 安装(一定要从源码安装)
官方的第一步也是bazel安装
这一步一定要进行配置,不然跑不了;这个可以根据自己实际的安装路径进行配置。
3.实现简单的c++调用tf的例子
copy里面的测试代码
- 训练tf模型,得到一个pb文件
- 根据文中的例子,形成以下目录;这里的cc文件和项目名称是一致的,可以不一致不过写BUILD文件的时候要注意了。
- 接下来,在tensorflow repo下(tensorflow的根目录下面),运行 ./configure 这一步就是用来设置一些基本的信息,可以参考别人的配置;也可以根据自己的实际情况。(这步在安装tensorflow已经完成可以不用做)
- bazel build //tensorflow/loader:loader 编译c++文件
- 从tensorflow的根目录进入到bazel-bin/tensorflow/my_loader文件中
- 把训练生成的pb文件拷贝到该目录下
- ./my_loader运行,查看结果