安装TensorFlow1.5+cuda9.0+cudnn7.0

发表于 | 分类于
字数统计: | 阅读时长 ≈

之前使用的一直是python2.7+TensorFlow1.1+cuda8.0+cudnn6.0,用了有一段时间了,python3是一直打算换过来的,毕竟py3比2增加了许多实用性的功能,而且py2逐渐不被许多库所支持,官方也声明在2020年放弃对py2的维护.

今天看到TensorFlow1.5版本的发布以及cuda9的发布想更新一波.

py

由于目前还会短暂使用py2,所以需要在一台电脑上同时使用py2和py3.

1
ps -aux | grep python

可以看到系统当前调用python的进程,看到python和python3都有.

可行的方法包括:

  1. virtualenv,不过暂时不想用这个.
  2. 最简单就是在用3的时候后面加3就好了, python3, pip3
  3. 也可以去pyenv看看

CUDA & CUDNN

官网下载cuda时选择9.0的版本,当前最新为9.1,刚开始安装了9.1的cuda toolkit,发现还是不支持,后来换到9.0版本就好了.

CUDNN也是官网下载deb包安装就可以.

驱动需要384以上,一个经验就是当电脑驱动出问题的时候,(经常登录界面无线循环,低分辨率等问题),最简单的解决办法就是把原来的使用linux apt安装的驱动包都卸载掉

1
sudo apt-get remove nvidia*

然后官网下载相应版本的driver, .run文件,用它来安装,一般可以解决驱动的问题.

TensorFlow1.5

本来打算源码安装的,进行到bazel build这一步的时候一个跟网络和bazel有关的问题解决不了,也没必要这么装了.

索性还是pip:

1
pip install tensorflow-gpu==1.5rc1 --user

这样原来的python2也是还可以继续用的,新的python3也ok.