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