本文转载自：https://my.oschina.net/u/3837179/blog/1920756

在ubuntu中配置GPU的深度学习环境相较于win问题要多很多，这几天琢磨了一下Ubuntu下的环境配置，参考很多人的博客，也遇到了不少坑，好不容易配置成功了，希望写下来，和大家分享，避免大家走弯路。环境的配置主要是nvidia显卡驱动的安装，在驱动安装的过程中遇到了问题，可以参考博客后面的问题解决方案，主要步骤就是装nvidia驱动，然后是安装cuda和cudnn，这两个一般问题不大。配置成功后就可以根据需要安装自己使用的深度学习框架（比如目前比较火的tensorflow和pytorch）。这两个框架本人建议使用Anconda安装，一来方便安装，二来如果出错，可以直接删除anconda文件夹重新安装即可。

软件要求

Ubuntu16.04

Nvidia gtx 750 Ti

Cuda 9.0

Cudnn 7.0

Nvidia显卡驱动的安装

1 禁用显卡驱动nouveau

网上很多文章说在官网下载run文件安装，这样安装的成功率较低，不建议这样做。还有在出现循环登录，黑屏时好多博客说卸载重装，这样也不大提倡，具体解决这些问题看博客后面的解决方案，亲测有效。由于nvidia的显卡驱动是闭源软件，因此ubuntu上默认是没有的，自带的显卡驱动叫做nouveau，安装之前要将nouveau先禁用掉，在终端里输入：

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

在vim编辑器中输入以下两行：

blacklist nouveau

options nouveau modset=0

然后配置文件生效，终端中输入：

sudo update-initramfs –u

2. 接下来安装一些后面会用到的32位lib，终端中依次输入

sudo apt-get install lib32ncurses5

sudo apt-get install lib32z1 

然后重启，输入：

sudo reboot

3. 再次进入系统

同时按下Ctrl+Alt+F1进入tty模式，登录账号，输入用户名密码。然后关闭图形界面服务，输入：

sudo service lightdm stop

接下来添加nvidia驱动的ppa源并进行安装：

sudo add-apt-repository ppa:graphics-drivers/ppa

查看可安装的驱动版本，选择合适的版本安装，这里安装的384版本，

ubuntu-drivers devices

安装nvidia驱动及其依赖的包，

sudo apt-get update sudo apt-get install nvidia-384

sudo apt-get install mesa-common-dev

sudo apt-get install freeglut3-dev

打开图形界面：

sudo service lightdm start

禁用Secure boot的问题(非常非常重要！！！好多博客说在BIOS中禁用Secure boot,这样也可以，但会影响到win的启动，所以直接使用mok工具即可，如果这一步操作不成功，就九九归一了，滑稽)：

sudo moktuil --disable-validation

然后输入8-16位密码（这里最好用简单的密码，本人之前设置很复杂的密码，结果在后面的mok manager界面输入密码时经常出错，所以如：88888888这样的密码很容易成功) 然后重启，输入：

sudo reboot

如果出现蓝色界面，选择改变secure boot状态的选项，进去提示输入对应位的密码（如:enter 2 characters:则输入密码的前两位），输入几遍后最后选择yes启动。

打开终端，输入，nvidia-smi和nvidia-settings如果输出如下图，说明安装成功：

这时nvidia就安装好了，如果碰到黑屏，循环登录，参考后面的问题解决方案。

Cuda的安装

找到下载文件的路径

sudo sh cuda_9.0.176_384.81_linux.run

回车，一路往下运行，直到提示“是否为NVIDIA安装驱动nvidia-384？”，选择否，因为已经安装好驱动程序了，其他的全都是默认，安装位置默认是/usr/local/cuda，配置环境变量，运行如下命令打开profile文件

sudo sh cuda_9.0.176_384.81_linux.run

打开文件后在文件末尾添加路径，也就是安装目录，命令如下：

export  PATH=/usr/local/cuda-9.0/bin:$PATH

export  LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH

保存，然后重启电脑

sudo reboot

测试CUDA的例子

cd  /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

如果显示的是关于GPU的信息，则说明安装成功了。 最后你会看到cuda驱动、sample、tookit已经安装成功,但是缺少一些库。

环境变量配置 安装完毕后，再声明一下环境变量，并将其写入到 ~/.bashrc 的尾部:

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}

exportLD_LIBRARY_PATH=/usr/local/cuda9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

然后设置环境变量和动态链接库，在命令行输入：

sudo gedit /etc/profile

在打开的文件末尾加入：

export PATH=/usr/local/cuda/bin:$PATH

保存之后，创建链接文件：

sudo gedit /etc/ld.so.conf.d/cuda.conf

在打开的文件中添加如下语句：

/usr/local/cuda/lib64

然后执行

sudo ldconfig

使链接立即生效

cudnn的安装

1) 继续Ctrl+alt+F1进入的字符界面 2) 进入解压后的cudnn-9.0-linux-x64-v7.0.tgz文件cuda，在终端执行下面的指令安装：

tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz

cd cuda

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

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

然后更新网络连接：

cd /usr/local/cuda/lib64/

sudo chmod +r libcudnn.so.7.0.3  # 自己查看.so的版本

sudo ln -sf libcudnn.so.7.0.3. libcudnn.so.7

sudo ln -sf libcudnn.so.7 libcudnn.so

sudo ldconfig

重新启动图形化界面

sudo service lightdm start

再Ctrl+alt+F7退出Text Mode

问题解决方案

在安装nvidia显卡驱动的过程中会遇到各种各样的问题，这里主要说明一下常见的3个问题：

问题1： 安装Ubuntu的时候，卡在Ubuntu的LOGO界面或黑屏 

问题2： 双显卡安装Nvidia驱动，循环登录或黑屏 

问题3： 启动后黑屏，并出现ubuntu the system is running in low-graphics mode 的错误

针对以上3个问题，我们这里提出了相应的解决方案。

注意！！！：对于问题1和问题2，如果下面对应的方法无效，则去掉”quiet splash”后面的apciosi或者nomodeset,在vthandoff后面添加acpi_osi或者nomodeset！！！

方案1（针对问题1）：

1. 启动系统后在Grub界面，选择ubuntu系统的那一行，然后按E键，就会进入Grub的编辑状态。

2、在quiet splash 后面加（先打空格）nomodeset，然后按F10保存启动

3、由于那个是临时改动，所以还需要一步，打开终端：

sudo vi /etc/default/grub

编辑打开的文件，找到GRUBCMDLINELINUX_DEFAULT那一行，在后面加上(在quiet splash后打一个空格) nomodeset，保存，然后在终端输入 ：

sudo update-grub

重启即可。

方案2（针对问题2）：

问题的根源是集显的问题（但由于我们是双系统或双显卡或着存在Grub启动项，导致没有解决到问题的根本！） 这一切的原因就是因为我们装系统的时候，加的那个参数nomodeset 接下来详细教程 1、 首先确定是否nvidia驱动已安装，按Ctrl+Alt+F1进入tty模式，登录，然后输入nvidia-smi如果输出显卡信息，说明安装成功了。 2、重新启动，在Grub界面选Ubuntu系统那一行然后按E键进入编辑模式：

3、也是上图那个位置，在 “quiet splash nomodeset”,主要看是否有nomodeset，有的话删除它（可能有人是queit splash= nomodeset,删除’= nomodeset’即可！） 4、然后在原来那个位置加 acpi_osi=linux（代码之间用空格隔开！），然后按F10启动，就可以进入界面了。

5、最后一步，保存信息，打开终端：

sudo vi /etc/default/grub 或 sudo gedit /etc/default/grub

#打开文件后将nomodest删除替换为 acpi_osi=linux

#然后更新grub

sudo update-grub

解释一下 Grub引导了系统进行启动，所以它的参数被传入了，即nomodeset（调用集显）如果存在，系统就会一直调用集显，然后就出现循环登录或黑屏。由于刚刚安装系统一般没有驱动，很多人只能通过调用集显去进入图形界面（除非在命令行下安装了驱动），导致了nomodeset参数的加入。 而acpi_osi=linux是告诉Grub，电脑将以Linux系统启动，调用其中驱动，所以可以用Nvidia的驱动进行显示了！

方案3（针对问题3）：

当开机出现ubuntu the system is running in low-graphics mode错误时，按下Ctrl+Alt+F1,登录

输入：

cd /etc/X11

然后输入：

sudo cp xorg.conf.failsafe xorg.conf

重启：

sudo reboot