安装NVIDIA驱动cuda和cudnn（以Ubuntu16为例）
一.安装NVIDIA驱动
- 根据自己的电脑的显卡版本，下载驱动文件，网站。
- 如果有NVIDIA驱动则卸载原有的NVIDIA驱动
sudo apt-get remove --purge nvidia*
- 禁用nouveau1
sudo gedit /etc/modprobe.d/blacklist.conf
#在文末添加以下两行，并保存、退出
blacklist nouveau
option nouveau modeset=0
#再执行以下命令
sudo update-initramfs -u
#重启后，输入以下命令若没有输出则说明禁用成功
lsmod | grep nouveau
- 开始安装NVIDIA驱动2
#输入以下命令后进入黑屏界面
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
- 输入网址https://developer.nvidia.com/cudnn找想安装的cudnn版本
- 按照官方文档https://docs.nvidia.com/deeplearning/sdk/cudnn-install/安装
- 输入以下命令，若显示有用信息则说明安装成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
来源：CSDN
作者：m0_37820219
链接：https://blog.csdn.net/m0_37820219/article/details/104060639