TensorFlow 1.8.0 + TensorRT 4.0 安装过程 (5月11日更新:目前在Ubuntu 18.04不可行)

在上一篇“基于tar文件的TensorRT 4.0安装过程”中,完成了TensorRT的安装,但由于其语言环境为C++,因而无法与TensorFlow结合,TensorFlow 1.7.0 已经集成了TensorRT,但需要从头编译安装,现记录一下编译安装过程。


首先从github上git TensorFlow源码:

$ git clone https://github.com/tensorflow/tensorflow.git 

安装bazel 0.10.0

配置configure:

$ cd tensorflow
# 切换tag
$ git checkout v1.8.0
$ ./configure 

出现的配置选项中只需要在CUDA support和TensorRT support两处选择“y”,其他支持选“n”即可,并输入正确的CUDA、cuDNN版本号和路径、TensorRT路径,我的环境对应的版本号分别为9.0、7.1,TensorRT路径为:/home/micro/TensorRT-4.0.0.3(注意,不要写成~/TensorRT-4.0.0.3,这里不能识别波浪符号)

编译

$ bazel build --config=cuda --config=opt //tensorflow/tools/pip_package:build_pip_package
# 编译过程较长,我这里跑了两个小时左右

生成安装文件

$ bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/
# 会在/tmp 目录下生成.whl文件

pip安装

$ sudo pip install sudo pip install /tmp/tensorflow-1.8.0-cp36-cp36m-linux_x86_64.whl
# tensorboard需单独安装
$ sudo pip install tensorboard
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值