Ubuntu16.04配置深度学习环境(TensorFlow-GPU)

时间:2021-07-24 17:22:20

一、安装NVIDIA驱动

去NVIDIA官网下载(http://www.nvidia.com/Download/index.aspx?lang=en-us)适合自己机器的GPU驱动(不用卸载系统自带的驱动也能成功)
如:NVIDIA-Linux-x86_64-384.111

二、安装驱动

1)先关闭图形环境,否则无法正常安装

命令:sudo service lightdm stop

然后按ctrl+alt+f1(f1-f6均可),输入Ubuntu用户名和登录密码

2)pcie bus error 报错解决

如果此时出现pcie bus error,则先退出安装(强制重启),重启之后在终端输入:sudo vim /etc/default/grub,编辑grub文件,将
GRUB_CMDLINE_LINUX_DEFAULT=”quiet splash” 改为GRUB_CMDLINE_LINUX_DEFAULT=”quiet splash pci=nomsi”,保存编辑;
然后输入:
sudo update-grub(更新grub文件),
再然后从1)开始操作;如果还是报pcie bus error,把pci=nomsi改为pci=noaer再试试。如果没有报错,进行第3)步;

3)安装驱动

cd 到驱动下载的目录,然后:sudo sh NVIDIA*.run(如果tab键不能补全的话,手动把驱动文件名敲出来),然后就一直accept或yes

4)sudo service lightdm start

5)sudo reboot

6)输入nvidia-smi查看驱动是否安装成功

三、安装cuda

先下载cuda(.deb文件,NVIDIA提供的安装教程使用的是runfile文件,但NVIDIA的安装教程中推荐用.deb文件)和补丁(即patch)(https://developer.nvidia.com/cuda-release-candidate-download

1)sudo dpkg -i cuda-repo-.deb

2)sudo apt-get update

3)sudo apt-get install cuda

4)安装补丁文件.deb

四、添加环境变量

输入命令:sudo vim ~/.bashrc
在文件最后添加:

Ubuntu16.04配置深度学习环境(TensorFlow-GPU)

重启系统:nvcc -V检查cuda是否安装成功

如果安装cuda之后想卸载重新装,卸载命令如下:
sudo apt-get –purge remove (输入cuda,然后按Tab键自动补全包名,可能包名补的不全,输入g(即带g的包名),再按Tab键就补全了)

五、安装cudnn

直接在百度上搜cudnn就会出来很多下载链接,下载cudnn时要先在NVIDIA上注册账号

先下载和cuda版本相匹配的cudnn,同一版本的cuda可能对应多个cudnn版本,这些不同版本的cudnn都可以用,但是会影响之后TensorFlow安装的版本选择
我的机器是Ubuntu16.04,cuda8.0,cudnn v6.0 for linux,TensorFlow-gpu 1.4.1

cd到cudnn的下载目录,然后在终端输入命令:

1)tar xvzf cudnn-8.0-linux-x64-v6.0.tgz(注意和自己下载的cudnn文件名一致)

2)cd cuda

3)按顺序敲下面的命令

sudo cp include/cudnn.h /usr/local/cuda/include

sudo cp lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

六、安装TensorFlow-GPU

本人比较懒,用pycharm装的TensorFlow,所以要先装pycharm(不会装pycharm去百度),由于Ubuntu16.04自带Python2.7和Python3.5,所以装好pycharm,直接指定解释器为系统自带的Python3.5就行了(不会指定pycharm解释器的,接着看),装好pycharm之后,打开pycharm,fille>settings>Project>Project Interpreter(这里可以选择解释器,指定为Python3.5),要装软件的话就点右侧边绿色的“+”,然后输入包名(如tensorflow-gpu),再点install,傻等十几分钟就装好了,装所有其它的Python库都在这里装,这么便捷的安装方法可能是pycharm如此强大的一个原因,最后要测试一下tensorflow-gpu装没装成,随便从网上找点tf的程序测试一下。

拿走不谢

[安装驱动参考] http://www.linuxidc.com/Linux/2017-10/147614.htm
[安装cuda和cudnn参考] http://blog.csdn.net/autocyz/article/details/52299889