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