ubuntu使用bazel编译安装tensorflow c++

按照官网步骤
1.下载安装cuda,cudnn(注意版本,我的是cuda9.0+cudnn7.5)
2.下载安装bazel,也需要注意版本
3.下载tensorflow源码https://github.com/tensorflow/tensorflow
4.进入tensorflow文件夹,

cd tensorflow
./configure

config按照提示选择,若有gpu,怎在选择cuda选项时选Y
5. bazel编译

bazel build --config=opt --config=cuda //tensorflow:libtensorflow_cc.so  #有显卡
bazel build --config=opt  //tensorflow:libtensorflow_cc.so #无显卡

编译需要等待较长时间
编译结束后,两个so库在bazel-bin/tensorflow文件夹下libtensorflow_cc.so 和 libtensorflow_framework.so
6. 安装其它依赖
在tensorflow/contrib/makefile下,执行build_all_linux.sh文件,成功后会出现一个gen文件夹
如果tensorflow/contrib/makefile下没有downloads文件夹,则回到tensorflow文件夹下,执行

./tensorflow/contrib/makefile/download_dependencies.sh

build_all_linux.sh 包含调用 download_dependencies.sh࿰

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值