hilinxinhui

Install Nvidia Driver on Ubuntu20.04

  1. 安装驱动
  2. 验证驱动安装情况
  3. cuda和cudnn

在Ubuntu 20.04上安装Nvidia显卡驱动。

  • 硬件平台：LEGION y7000p 2020
  • 显卡型号：NVIDIA RTX 2060 6G（notebook）
  • 驱动版本：535.54.03

安装驱动

在这里选择对应的驱动版本并下载。安装指引请看这里。

tl;dr:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# 卸载原有驱动（若有）
$ sudo apt remove --purge nvidia*

# 禁用nouveau
$ sudo vim /etc/modprobe.d/blacklist.conf
# 填入 blacklist nouveau
# 填入 options nouveau modeset=0
$ sudo update-initramfs -u # 更新内核

# 重启后验证nouveau已禁用
$ lsmod | grep nouveau

# 关闭图形化界面，这一步不是必须的
$ sudo service lightdm stop
# Ctrl + Alt + F3，因机而异

# 安装驱动
$ sudo chmod +x NVIDIA-Linux-x86_64-535.54.03.run # 以此版本驱动为例
$ sudo ./NVIDIA-Linux-x86_64-535.54.03.run # 以此版本驱动为例

# 开启图形界面，同理，这一步不是必须的
$ sudo service lightdm start
# Ctrl + Alt + F2，因机而异

安装过程遵照提示，作者遇到了以下两个问题：

  • 需要sudo权限执行安装脚本
  • 需要安装make进行编译

按照错误信息重新执行/安装需要的东西即可。

验证驱动安装情况

1
$ nvidia-smi

命令返回：

  • 驱动版本信息、兼容的cuda版本信息
  • 显卡型号、显存使用情况和正在使用显卡的进程

nvidia_smi

cuda和cudnn

作者一般用conda安装cudatoolkit（或cuda-runtime一类的东西）和cudnn（更方便版本管理）。安装命令看起来像这样（版本可能不一样）：

1
$ conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
1
2
3
import torch

print(torch.cuda.is_available()) # True
Dev, DL NVIDIAUbuntu