安装mxnet+显卡驱动+cuda9.1+cudnn7.1
操作系统：ubuntu16.04
GPU：GTX1080
一、安装显卡驱动
1. 首先查看GPU版本信息
$ lspci | grep -i nvidia
2. 验证自己的Linux版本是否支持 CUDA(Ubuntu 16.04和Ubuntu17.04都支持)
$ uname -m && cat /etc/*release
3. 验证是否安装gcc
$ gcc --version
如果没有则输入 $ sudo apt-get install gcc下载
4. 验证系统是否安装了kernel header和 package development
$ uname -r
$ sudo apt-get install linux-headers-$(uname -r)
5. 卸载已安装驱动
$sudo apt remove --purge nvidia*
6.禁用nouveau驱动
在终端运行 $ lsmod | grep nouveau
如果有输出，说明nouveau正在运行，需要手动禁用
在/etc/modprobe.d中创建文件blacklist-nouveau.conf
输入 $sudo vim /etc/modprobe.d/blacklist-nouveau.conf
在文件中输入一下内容：
blacklist nouveau
options nouveau modeset=0
执行
$ sudo update-initramfs -u
查看nouveau是否已经成功禁用
$ lsmod | grep nouveau
然后$ sudo reboot重启
重启后进入用户登录界面时（不要登录进桌面），按Ctrl+Alt+F1进入字符终端界面，输入用户名和密码进行登录，登录成功后，再次查看。
关闭图形界面
$ sudo service lightdm stop
7.安装驱动
这里采用apt安装驱动
添加Graphic Drivers PPA
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt-get -update
安装nvidia driver,我安装的驱动版本是387，之前装的nvidia-384在mxnet跑程序时出现错误，提示版本太低。
输入 $ sudo apt-get install nvidia-387
安装完成后重启
$ sudo reboot
重启后可以执行$ sudo nvidia-smi 查看驱动版本和显卡型号
二、安装cuda
在英伟达官网上下载cuda9.1, https://developer.nvidia.com/cuda-downloads
按图中的选择下载即可
在/home/downloads里面可以看到下载的cuda文件
执行 $ sudo sh cuda9.1.85_387.26_linux.run
之后就是很长的一段说明，一直按回车即可。
当提示“是否为NVIDIA安装驱动？”时选择“N”，不然会把之前的驱动覆盖掉。其他选择默认即可。
设置环境变量
$ sudo vim ~/.bashrc 来编辑.bashrc配置文件
在文件最后加上下面几句话（直接复制就行）
$ export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
$ export LD_LIBRARY_PATH=/usr/local/cuda8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
$ export CUDA_HOME=/usr/local/cuda
然后设置环境变量和动态链接库，在命令行输入：
$ sudo vim/etc/profile
在打开的文件末尾加入：
export PATH = /usr/lacal/cuda/bin:$PATH
保存之后，创建链接文件
$ sudo gedit /etc/ld.so.conf.d/cuda.conf
在打开的文件中添加如下语句：
/usr/local/cuda/lib64
然后执行
$ sudo ldconfig
重启一下机子
三、安装cudnn
首先在官网上下载 https://developer.nvidia.com/rdp/form/cudnn-download-survey
进去要注册登录
下载第一个for CUDA9.1的
首先解压tgz文件
$ sudo -xzvf cudnn-9.1-linux-x64-v7.tgz
然后复制文件到cuda安装路径下
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
即可完成cudnn安装
如果提示库缺失错误，可参考以下命令：
$ sudo cp /usr/local/cuda-9.1/lib64/libcudart.so.9.1 /usr/local/lib/libcudart.so.9.1 && sudo ldconfig
$ sudo cp /usr/local/cuda-9.1/lib64/libcublas.so.9.1 /usr/local/lib/libcublas.so.9.1 && sudo ldconfig
$ sudo cp /usr/local/cuda-9.1/lib64/libcurand.so.9.1 /usr/local/lib/libcurabd.so.9.1 && sudo ldconfig
$ sudo cp /usr/local/cuda-9.1/lib64/libcudnn.so.7 /usr/local/lib/libcudnn.so.7 && sudo ldconfig
新版的cudnn下载变成了deb文件了
所以直接在终端输入：
sudo dpkg -i libcudnn7_7.3.1.20-1+cuda9.0_amd64.deb
即可完成对cudnn的安装，非常简单。