用C++导入tensorflow模型环境配置

1.安装bazel,根据官方教程进行安装

  • 注意版本号
  • 注意安装的方式,这里有两种安装方式,第一种安装方法一般是安装最新版本,第二种则是手动下载所需的版本进行安装bazel version(测试安装完成!)

2.Tensorflow 安装(一定要从源码安装)

官方的第一步也是bazel安装


用C++导入tensorflow模型环境配置_第1张图片
官网截图
用C++导入tensorflow模型环境配置_第2张图片
配置configure
用C++导入tensorflow模型环境配置_第3张图片
配置configure

这一步一定要进行配置,不然跑不了;这个可以根据自己实际的安装路径进行配置。

用C++导入tensorflow模型环境配置_第4张图片
安装tensorflow

3.实现简单的c++调用tf的例子

copy里面的测试代码

  • 训练tf模型,得到一个pb文件
  • 根据文中的例子,形成以下目录;这里的cc文件和项目名称是一致的,可以不一致不过写BUILD文件的时候要注意了。
用C++导入tensorflow模型环境配置_第5张图片
目录
  • 接下来,在tensorflow repo下(tensorflow的根目录下面),运行 ./configure 这一步就是用来设置一些基本的信息,可以参考别人的配置;也可以根据自己的实际情况。(这步在安装tensorflow已经完成可以不用做)
  • bazel build //tensorflow/loader:loader 编译c++文件
  • 从tensorflow的根目录进入到bazel-bin/tensorflow/my_loader文件中
  • 把训练生成的pb文件拷贝到该目录下
  • ./my_loader运行,查看结果

你可能感兴趣的:(用C++导入tensorflow模型环境配置)