最近测试大模型的时候要用到GPU,今天把GPU驱动记录如下。如有问题,欢迎评论区留言。
系统环境:ubuntu server 22.04
1、安装Nvidia驱动1.1 加入官方ppa源:
add-apt-repository ppa:graphics-drivers/ppa
1.2 apt update
apt-get update
apt-get list --upgradable
apt-get upgrade
1.3 安装Neofetch
apt install neofetch
apt install lolcat
1.4 查看硬件信息
neofetch && uname -a|lolcat
1.5 查看GPU适配的驱动版本: ubuntu-drivers devices 可以看到建议的驱动版本,我这里的GPU是RTX 3060
1.6 安装驱动
apt install nvidia-driver-545
1.7 disable the nouveau
vim /etc/modprobe.d/blacklist.conf
#在最后加入(也可以只加第一行)
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
#更新系统
update-initramfs -u
reboot #必须重启
#用命令nvidia-smi 来查看: