1. GPU 驱动升级

[1] - 添加 ppa，安装 GPU 驱动：

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-390 # 安装最新驱动的版本号

[2] - 安装 cuda toolkit：

sudo apt install nvidia-cuda-toolkit

[3] - 重启电脑

sudo reboot

2. CUDA 升级

如 CUDA 10.0 升级到 CUDA10.1：

[1] - 卸载 CUDA 10.0：

sudo /usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl
#如果有cudnn，
sudo rm -rf /usr/local/cuda-10.0

[2] - 安装 CUDA 10.1：

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run