安装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下载

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的安装，非常简单。