ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

Ubuntu安装编译TensorFlow C++

2021-12-29 21:03:10  阅读:397  来源: 互联网

标签:protobuf 编译 -- Ubuntu C++ bazel tensorflow TensorFlow


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

标签:protobuf,编译,--,Ubuntu,C++,bazel,tensorflow,TensorFlow
来源: https://blog.csdn.net/MclarenSenna/article/details/122217170

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有