深度学习环境——安装NVIDIA驱动、cuda和cudnn（以Ubuntu16为例）

一.安装NVIDIA驱动

1. 根据自己的电脑的显卡版本，下载驱动文件，网站。
2. 如果有NVIDIA驱动则卸载原有的NVIDIA驱动

sudo apt-get remove --purge nvidia*

3. 禁用nouveau¹

sudo gedit /etc/modprobe.d/blacklist.conf
#在文末添加以下两行，并保存、退出
blacklist nouveau
options nouveau modeset=0
#再执行以下命令
sudo update-initramfs -u
#重启后，输入以下命令若没有输出则说明禁用成功
lsmod | grep nouveau

4. 开始安装NVIDIA驱动²

#输入以下命令后进入黑屏界面
sudo service lightdm stop
按Ctrl+Alt+F1进入命令行
进入NVIDIA文件的下载目录
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.xxx.run
sudo ./NVIDIA-Linux-x86_64-xxx.xx.run -no-opengl-files
#回到图形界面
sudo service lightdm start
#完成后在终端输入以下命令，若显示驱动的信息则安装成功
nvidia-smi

二.安装cuda

搜索并进入cuda官网，下载cuda，按照官方文档Online Documentation安装。
但是官方文档给的添加环境变量还需要完善一下，应该要设置成永久有效的。

sudo vim /etc/profile
#在最后添加两行
export PATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
#使它们生效
source /etc/profile 

三.安装cudnn

1. 输入网址https://developer.nvidia.com/cudnn找想安装的cudnn版本
2. 按照官方文档https://docs.nvidia.com/deeplearning/sdk/cudnn-install/安装
3. 输入以下命令，若显示有用信息则说明安装成功

 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2    

四.卸载cuda

sudo apt-get remove cuda
sudo apt-get autoclean
sudo apt-get remove cuda*
cd /usr/local/
sudo rm -r cuda-9.0

五.删除cudnn

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

---

1. https://blog.youkuaiyun.com/zhuiqiuzhuoyue583/article/details/88781293. ↩︎

2. https://blog.youkuaiyun.com/zhang970187013/article/details/81012845 ↩︎