一、安裝英偉達 GPU 驅動
#安裝ubuntu後運行以下命令來升級內核版本
sudo apt-get update
sudo apt-get upgrade
#以下命令會將與你係統相兼容的驅動版本顯示出來
sudo add-apt-repository ppa:graphics-drivers/ppa
#安裝所需版本的NVIDIA驅動
sudo apt-get update
sudo apt-get install nvidia-<driver_number>
#比如你的版本爲410則指令如下
sudo apt-get update
sudo apt-get install nvidia-410
#記得安裝完NVIDIA驅動要重啓
sudo shutdown -r now
#驗證NVIDIA驅動是否裝好
nvidia-smi
#看是否出現如下界面
二、 安裝 CUDA和CUDNN
接下來參照該教材即可:https://blog.csdn.net/wanzhen4330/article/details/81699769
CUDA 9.0