升级Nvidia驱动

1. 使用 PPA 第三方软件仓库安装最新版本

sudo add-apt-repository ppa:graphics-drivers/ppa

2. 更新软件源

sudo apt update

3. 查看显卡型号和推荐驱动

ubuntu-drivers devices

• 如果同意安装推荐的驱动，可以直接运行：

sudo ubuntu-drivers autoinstall

• 也可以直接指定希望安装的驱动：

sudo apt install nvidia-drivers-xxx

sudo sed -i -e 's/bionic/xenial/g' /etc/apt/sources.list

---

Note: 如果是想从低版本升级或安装其他版本，需要先卸载原有的的驱动：

sudo apt-get purge nvidia*

二. 系统安装CUDA和cuDnn

1. 这里以CUDA10.2安装为例。
   进入官网页面后，根据硬件和系统版本，页面会给出合适的配置和指示，如下图所示：

image.png

2. cuDNN安装

• 下载 https://developer.nvidia.com/rdp/cudnn-download
• 安装命令

sudo dpkg -i libcudnn8_8.0.5.39-1+cuda10.2_amd64.deb

Note: pytorch条件下可以选择不在系统安装CUDA和cuDnn，其内部自带cuda和cudnn的安装。在安装了正确的Nvidia驱动后，可以随着pytorch一起安装在anaconda虚拟环境中，非常方便。

Ref

https://zhuanlan.zhihu.com/p/106708516