参考:https://blog.csdn.net/zataji/article/details/123104569、https://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
,则说明安装成功。