1、下载TensorFlow源码
https://github.com/tensorflow/tensorflow
2、安装编译protobuf
protobuf版本要和TensorFlow版本对应;
protobuf版本查看方法:
打开TensorFlow源码下tensorflow-master/tensorflow/workspace2.bzl文件
搜索可以找到地址"https://github.com/protocolbuffers/protobuf/archive/v3.9.2.zip"
安装编译:
cd protobuf-XXX/
./autogen.sh
./configure
make
make check
sudo make install
sudo ldconfig # refresh shared library cache.
查看protobuf版本
protoc --version
3、安装编译bazel
bazel版本要和TensorFlow版本对应;
bazel版本查看方法:
打开TensorFlow源码下tensorflow-master/configure.py
搜索可以找到:
_TF_MIN_BAZEL_VERSION = '4.2.2'
_TF_MAX_BAZEL_VERSION = '4.99.0'
这里可以选择安装4.2.2版本
安装编译:
下载:链接 Releases · bazelbuild/bazel · GitHub
安装依赖:
sudo apt-get install pkg-config zip g++ zlib1g-dev unzip python
编译:
chmod +x bazel-<version>-installer-linux-x86_64.sh
./bazel-<version>-installer-linux-x86_64.sh --user
4.2.2版本对应写法:
chmod +x bazel-4.2.2-installer-linux-x86_64.sh
./bazel-4.2.2-installer-linux-x86_64.sh --user
配置环境:
sudo gedit /etc/bash.bashrc
将 export PATH="$PATH:$HOME/bin" 添加进去
保存:source /etc/bash.bashrc
4、编译TensorFlow
进入tensorflow文件夹中,首先进行项目配置:
./configure # 如果只需要配置cpu环境就一直回车
注:如果选用cpu,只需要一直点确定继续就好,都是选择no;
使用bazel编译TensorFlow:
bazel build --config=opt //tensorflow:libtensorflow_cc.so
编译完成后,在bazel-bin/tensorflow中会生成两个我们需要的库文件:libtensorflow_cc.so 和 libtensorflow_framework.so
期间会因为网络了连接不上github,多尝试几次就好。
参考:Ubuntu安装TensorFlow C++_蓬莱道人的博客-CSDN博客
ubuntu18.04 tensorflow以及tensorflow lite源码编译C++库_上善若水-CSDN博客_ubuntu 编译tensorflowlite