Ubuntu16.04 / Windows10 安装 Tensorflow GPU版教程 CUDA9.0 - 请转CDA8.0

时间:2021-06-06 13:50:45

注意这几点,少填坑:

安装顺序:Python -> Tensorflow -> Nvidia GPU Driver -> CUDA -> CUDNN

其中,可以在安装完成TensorFlow之后,尝试在python中import tensorflow,然后看给出的错误信息,其中包含所需要的CUDA和CUDNN版本,比如:

1
ImportError: libcudnn.so.6: cannot  open  shared object  file : No such  file  or directory

表示需要CUDNN6.0版本,而与其对应的是CUDA8.0。

一定要下载对应的CUDA和CUDNN。

Tensorflow不支持CUDA9.0,故请将以下所有涉及9.0的换成8.0即可。

下面的内容可以不用看了,仅供参考。

建议看这一篇:http://www.cnblogs.com/xuliangxing/p/7575586.html

=====================

Windows10中,下载CUDA 8.0之后,在安装的时候选择自定义安装,并去掉GEFORCE和驱动,否则会安装失败。

安装完成之后,在python中调用tensorflow发现缺少cudnn6_64.dll,这意味着需要下载cudnn6 for cuda8.0。

下载完了直接将文件放到对应的文件夹中即可。

再次调用tensorflow,成功。

=====================

最终成功:

CUDA和CUDNN的测试也通过了,就不放图了。放一张import tensorflow成功的。

Ubuntu16.04 / Windows10 安装 Tensorflow GPU版教程 CUDA9.0 - 请转CDA8.0

 

Ubuntu16.04 / Windows10 安装 Tensorflow GPU版教程 CUDA9.0 - 请转CDA8.0

==============================

1、下载CUDA 9.0

从官网下载CUDA9.0,注意,如果网速太慢,不能10分钟之内下载完毕,最好用虚拟主机或者百度网盘离线下载。

我用本机下载尝试了好几次,都失败了,最后用的是Azure上的Ubuntu下载的,接着用/var/www/html分享,此外还有NFS、SSH、FTP等方法,可以在本地共享远程的ubuntu文件。Azure上的磁盘是托管的,如果删除了虚拟主机,应该可以直接全部导出磁盘上的文件,不过我没有试过。

Linux上的常用下载软件是uget,用uget代理firefox的全部下载链接。可以先用uget从官网下载,失败之后千万不要重试,让其hang out,然后再从虚拟主机或者其他的链接用uget下载,这时uget会自动续传。

 2、安装NVIDIA显卡驱动

这个时候,直接在tty1中操作,请确保有网络链接。

1
sudo  service lightdm stop  #关闭图形界面

黑屏、开机时候的样子,接着按 ctrl+alt+F1进入tty1,登录。

 

我们发现下载的CUDA文件名中有384字样,这意味着我们需要下载nvidia-384这个驱动。为了避免出现循环登录、“Low graphical mode”以及不识别各种硬件的问题,我们需要先将默认的若干显卡驱动添加到一个配置文件中:

1
2
3
sudo  apt-get  install  vim   #自带的vi是屎
cd  /etc/modprobe .d/
sudo  vim blacklist.conf

 并且在最后添加:blacklist nouveau

1
sudo  update-initramfs -u

 接着直接安装nvidia-384

1
2
sudo  apt-get remove --purge nvidia-*  #先卸载已经安装的驱动
sudo  apt-get  install  nvidia-384 nvidia-xconfig nvidia-settings nvidia-prime 
1
sudo  update-initramfs -u

3、安装CUDA9.0

依然实在tty1中进行,找到下载的CUDA文件,参照官网在下载按钮下方给出的安装方法,输入命令即可。我的是.run文件,因此

1
sudo  sh cuda_9.0_file.run

按空格快速浏览协议,accept之后,不用安装cuda中带的384驱动,输入n。接着输入y或者按回车。

安装完毕之后,重启。

1
sudo  reboot

至此,CUDA安装完毕。新安装的显卡驱动对桌面的支持不好,在拖动窗口的时候会有锯齿。╮(╯▽╰)╭(我自己安装的时候都是坑。)

 

安装Tensorflow-gpu

1
sudo  pip3  install  tensorflow-gpu

一定要用sudo,否则白下。