准备跑老师给定的Github上的多模态源码,但是用了这么久ubuntu还没有尝试过安装nvidia驱动,好在也是一次成功,于是记录下来。
借鉴的是https://blog.csdn.net/Eric_xkk/article/details/131800365这篇文章,按照流程来基本没有问题,不过个人觉得有些步骤比较冗余,所以记录下来
主要流程
关闭安全模式
这个在安装双系统的时候就在window中关闭过了,所以不设置
打开独显模型
这个没有设置,但是最后也可以成功识别到显卡
更新软件源
sudo apt-get update #更新软件列表sudo apt-get install g++sudo apt-get install gccsudo apt-get install make
禁用nouveau
nouveau是系统自带的驱动程序,一般来讲要安装新的显卡驱动的时候需要禁用这个驱动
首先打开nouveau的配置文件:
sudo gedit /etc/modprobe.d/blacklist.conf
然后在最下面加上:
blacklist nouveau
options nouveau modeset=0
保存修改之后在终端执行:
sudo update-initramfs –u
重启电脑
终端执行:
sudo reboot
(lightdm那一步个人觉得可做可不做,好像和默认的gdm3的区别就是显示占用的内存大小,但是我看了一下lightdm占用的是6MB左右,gdm3占用的是9MB作用,区别不大。按照个人的喜好设置即可,其实就是一个显示器插件)
安装显卡驱动工具
sudo apt-get install nvidia-cuda-toolkit
查看系统推荐显卡驱动
这里要看好哪个是系统推荐的驱动(recommend),并且要记下来
sudo ubuntu-drivers devices
比如我的系统推荐的就是550(很奇怪,当时第一次下载的时候推荐的是550,现在是560...不过区别应该不大)
这里说一下三种驱动的大致区别:
- nvidia-driver-xxx :这是适配大多数用户,一般情况下都可以安装这个
- nvidia-driver-xxx-server :这个适配服务器,有多块GPU调度等情况
- nvidia-driver-xxx-open :这个是开源版本,可以自己修改
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd000028A0sv0000103Csd00008BBEbc03sc00i00
vendor : NVIDIA Corporation
driver : nvidia-driver-545 - distro non-free
driver : nvidia-driver-545-open - distro non-free
driver : nvidia-driver-550-open - third-party non-free
driver : nvidia-driver-565 - third-party non-free
driver : nvidia-driver-560-open - third-party non-free
driver : nvidia-driver-550 - third-party non-free
driver : nvidia-driver-560 - third-party non-free recommended
driver : nvidia-driver-535-open - distro non-free
driver : nvidia-driver-535-server-open - distro non-free
driver : nvidia-driver-535-server - distro non-free
driver : nvidia-driver-565-open - third-party non-free
driver : nvidia-driver-535 - distro non-free
driver : xserver-xorg-video-nouveau - distro free builtin
添加驱动源
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
下载驱动
在ubuntu的 软件和更新 中的 附加驱动中找到刚才系统recommend的显卡驱动型号,然后下载
测试
安装之后reboot重启电脑,输入nvidia-smi
查看信息,如果出现下列信息,则说明安装成功:
Tue Dec 24 12:12:34 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 550.142 Driver Version: 550.142 CUDA Version: 12.4 |
|-----------------------------------------+------------------------+----------------------+
| GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|=========================================+========================+======================|
| 0 NVIDIA GeForce RTX 4060 ... Off | 00000000:01:00.0 Off | N/A |
| N/A 44C P0 588W / 80W | 9MiB / 8188MiB | 0% Default |
| | | N/A |
+-----------------------------------------+------------------------+----------------------++-----------------------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=========================================================================================|
| 0 N/A N/A 1735 G /usr/lib/xorg/Xorg 4MiB |
+-----------------------------------------------------------------------------------------+
修改界面登陆器
如果前面下载了lightdm,觉得不好看想换回gdm3(比如我),输入sudo dpkg-reconfigure lightdm
后选中gdm3即可,reboot重启之后即可奏效
cuda检查
在pycharm中输入torch.cuda.is_availiable()
,输出为true,说明安装成功,大功告成!