ubuntu下极简安装tensorflow-gpu教程-不用手动安装cuda和cudnn

版权声明:转载请标注来源 https://blog.csdn.net/hehedadaq/article/details/82900933
  1. 安装anaconda,换成中科大源,网上随便找!
  2. 创建一个虚拟环境:conda create -n gtf matplotlib numpy pillow yaml python=2.7
  3. gtf 为虚拟环境的名字,后面的是需要装的包,Python可以选2.7也可以选3.5 ,直接安装就好了
  4. 激活环境:source activate gtf
  5. 安装tensorflow-gpu: conda install tensorflow-gpu==1.6.0
  6. 静等安装好,你会发现,咦,conda list里面有cuda-toolkit 和cudnn!,而且是9.0和7.12版本的!
  7. 清华源只有8.0版本的

这里还不够,还是有bug的,你运行的时候会发现,一个是导入不了tensorflow,或者是Session有问题。
具体报错,我忘了,等我下次重装系统再加上。
反正问题就是显卡驱动没有装
我室友推荐的是PPA安装,我找到的教程如下:
ubuntu16.04下NVIDIA GTX965M显卡驱动PPA安装
主要是在终端输入下面的几步:

  1. 找到你匹配的显卡驱动:
  2. sudo add-apt-repository ppa:graphics-drivers/ppa
  3. sudo apt-get update
    找到适合你的本子的驱动版本:
  4. ubuntu-drivers devices

在这里插入图片描述

大概就是最后一个,选择那个recommended,推荐的那个,然后输入:
5. sudo apt-get install nvidia-378
不用进入tty模式好像也行。反正我就没进入,你要是不成的话,和原教程里走一遍就行。
最后是重启

sudo reboot

重启系统后,执行下面的命令查看驱动的安装状态显示安装成功

sudo nvidia-smi
sudo nvidia-settings

应该还是有一个opencv的问题,如果是Ubuntu14的话,Python2.7的话,需要安装opencv-python

pip install opencv-python

这个比较慢,大家最好是在pypi官网上找到你需要的版本,直接下,然后离线安装:
就是进入文件目录,然后pip install name

结束

基本上就OK了,不用去考虑复杂的cuda下载,安装以及版本不匹配的问题。整个流程下来,大概也就几个小时,关键是bug少!

更新:Windows下免安装cuda和cudnn。

同样只能是中科大源,你更新完了,用这个语句,看看是不是,还得更新一下,所以依次输入下面两个命令:
conda config --set show_channel_urls yes
conda config --show channels
显示的只有中科大的才行!
然后根据最新更新的源,1.6.0的TensorFlow-GPU版的Windows没有了,只能安装1.8.0的了。
具体的可以看看这个链接:
https://mirrors.ustc.edu.cn/anaconda/pkgs/main/win-64/
然后ctrl+f,搜搜tensorflow-gpu,看看最低的版本,
然后搜搜cudatoolkit,看看有没有,自己随意吧~
这个也是需要安装显卡驱动的,这个我之前安装过,所以就没教程了,大家自己找找吧~

-----2018年10月20日22:30

猜你喜欢

转载自blog.csdn.net/hehedadaq/article/details/82900933