版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Allyli0022/article/details/80507476
1、查看显卡信息：
lspci | grep -E "VGA|3D" （或者lspci | grep -i nvidia）
2、查看可以安装的驱动：
ubuntu-drivers devices
3、假如看到nvidia-361驱动可用，忽略下面的nouveau（可以加入黑名单），接着使用如下命令安装NVIDIA驱动、N卡的设置面板以及双显卡切换工具：
sudo apt-get install nvidia-361 nvidia-settings nvidia-prime
4、重新启动：
reboot
5、安装cuda包：
sudo apt-get install nvidia-cuda-toolkit
当使用ubuntu18.04的时候，安装的是cuda9.1
6、安装cudnn:
拷贝一份解压后的cudnn到宿主机器的/home下，然后如下：
cp -f cuda/include/cudnn.h /usr/local/include/
cp -f cuda/lib64/libcudnn* /usr/local/lib
chmod 777 /usr/local/include/*
chmod 777 /usr/local/lib/*
7、建立软连接：
sudo ln -sf libcudnn.so.5.10 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
8、测试：
nvidia-smi
如果出现表格信息，则驱动安装成功。
nvcc -V
如果出现cuda版本信息，则cuda安装成功。
参考：https://blog.csdn.net/Real_Myth/article/details/52288635?locationNum=8&fps=1