Ububtu下使用源码安装Tensorflow GPU版本

截至2017年12月只有在安装了CUDA toolkit 8.0和CuDNN v6的64位Ubuntu下可以通过pip安装支持GPU的TensorFlow,对于其他的Linux版本和其他的CUDA/CuDNN版本的用户,需要通过源码进行安装来支持GPU。

选择Tensorflow版本

安装Tensorflow前先确认自己需要的版本,确定使用的版本后再往下进行。
由于需要使用bazel编译Tensorflow源码,需要CUDA和CUDNN实现GPU加速,所以确认Tensorflow版本后,还要确认bazel、CUDA、CUDNN版本。
部分版本配合结果如下
linux:
version比如此时需要在python27下使用bazel安装tensorflow_gpu-1.12.0,相应的我们选择bazel版本为0.15.0,CUDA版本为9.1,CUDNN版本为7.1。

安装bazel 0.15.0

CUDA 和CUDNN安装详见CUDA安装、CUDNN安装
在安装 bazel 时,要先安装 JDK8(按照官方网站安装就没问题:https://docs.bazel.build/versions/master/install.html)

  1. 安装JDK 8

sudo apt-get install openjdk-8-jdk

  1. 下载bazel 0.15.0
    地址:bazel
    将下载好的文件复制到HOME目录下,赋予权限:

chmod +x bazel-0.15.0-installer-linux-x86_64.sh

  1. 安装

./bazel-0.15.0-installer-linux-x86_64.sh --user

完成后会在HOME目录下生成一个bin文件夹

  1. 查看bazel 版本

bazel version

bazel version
从上图可以看到,bazel版本是0.15.0。

编译安装Tensorflow

  1. 下载Tensorflow

git clone https://github.com/tensorflow

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值