参考:https://blog.csdn.net/zataji/article/details/123104569https://blog.csdn.net/choimroc/article/details/104630491#:~:text=WARNING%3A Unable to determine the path to install,libraries installed%2C or specify a path with --glvnd-egl-config-path.

# 一、禁用 nouveau 驱动

因为 NVIDIA 的官方驱动和系统自带的 nouveau 驱动冲突,所以需要先禁用。

编辑文件 /etc/modprobe.d/blacklist-nouveau.conf

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

并在文件末尾添加以下内容:

blacklist nouveau
options nouveau modeset=0

执行以下命令使禁用生效并且重启

sudo update-initramfs -u
# 重启
sudo reboot

重启后验证是否生效, 执行命令,若没有输出,则禁用生效

lsmod | grep nouveau

# 二、安装 NVIDIA 驱动

根据显卡的参数到 NVIDIA 官网查询显卡匹配的驱动版本。目前所使用惠普电脑所使用的显卡型号为 GeForce RTX 3050。

下载好驱动,执行命令进行安装。

sudo su
sudo sh NVIDIA-Linux-x86_64-550.120.run

# 三、验证

输入命令 nvidia-smi ,验证是否安装成功,如果安装成功,会出现显卡的相关信息

root@ember-Victus-by-HP-Laptop:/home/ember/下载# nvidia-smi
Tue Oct 22 01:17:32 2024       
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 550.120                Driver Version: 550.120        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 3050 ...    Off |   00000000:01:00.0 Off |                  N/A |
| N/A   44C    P0             13W /   60W |       1MiB /   4096MiB |      6%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+
                                                                                         
+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|  No running processes found                                                             |
+-----------------------------------------------------------------------------------------+

在 python 中执行

import torch
print(torch.cuda.is_available())

程序输出 True ,则说明安装成功。

更新于 阅读次数