一、安装英伟达 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.youkuaiyun.com/wanzhen4330/article/details/81699769
CUDA 9.0