准备跑老师给定的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 gcc
 
sudo 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,说明安装成功,大功告成!

标签: none

添加新评论