暂无图片
暂无图片
暂无图片
暂无图片
暂无图片

搭建ai模型开发环境-英伟达显卡驱动安装

IT那活儿 2025-05-13
688

点击上方“IT那活儿”公众号--专注于企业全栈运维技术分享,不管IT什么活儿,干就完了!!!


驱动下载

Step1 下载显卡驱动

wget https://cn.download.nvidia.com/XFree86/Linux-x86_64/550.142/NVIDIA-Linux-x86_64-535.54.03.run

卸载可能存在的旧版本 nvidia 驱动(对于新机器或者从未安装过nvidia显卡驱动的机器,这步可以省略,但推荐执行

sudo apt-get remove --purge nvidia*

sudo apt-get --purge remove nvidia*

sudo apt-get autoremove

sudo apt-get --purge remove"*cublas*""cuda*"

sudo apt-get --purge remove "*nvidia*"

Step2 执行命令

如果执行了以上步骤,输入命令nvidia-smi仍可显示驱动信息(一般情况只卸载显卡驱动使用以下命令,适用于.run文件安装的驱动),再执行以下命令:

/usr/bin/nvidia-uninstall

Step3 禁用nouveau

修改文件 etc/modprobe.d/blacklist-nouveau.conf,最后添加如下两行:

blacklist nouveau
 
options nouveau modeset=0

#修改文件 /etc/modprobe.d/blacklist.conf,最后添加如下四行:

blacklist nouveau
blacklist nouveau modeset=03
blacklist nouveau
blacklist nouveau modeset=0

Step4 保存后,执行如下命令更新改动

sudo update-initramfs -u

Step5 执行sudo reboot

重启电脑,nouveau驱动应该已被禁用,执行下面命令没有输出才对:

lsmod | grep nouveau

直到命令不输出任何内容:

sudo dpkg --list | grep nvidia-*
 
sudo dpkg --list | grep nvidia*


驱动安装前准备工作

2.1 更新软件列表

sudo apt-get update

2.2 安装编译依赖

sudo apt-get install g++
 
sudo apt-get install gcc
 
sudo apt-get install make


 驱动安装

第一种安装方式 :自动化安装显卡驱动

查看支持的显卡:

ubuntu-drivers devices

据显示的推荐结果选一个,我选的是535版本,一般选择后面跟有recommend字样的版本:

apt install nvidia-driver-535

或者使用自动安装显卡指令(我采用此种)

sudo ubuntu-drivers autoinstall

第二种安装方式:下载deb文件

dpkg -i nvidia-driver-local-repo-ubuntu2004-535.54.03_1.0-1_amd64.deb

按提示复制:

apt-get update
 
apt-get install nvidia-driver-535 -y

查看先看是否识别:

nvidia-msi


END


本文作者:刘 营(上海新炬中北团队)

本文来源:“IT那活儿”公众号

文章转载自IT那活儿,如果涉嫌侵权,请发送邮件至:contact@modb.pro进行举报,并提供相关证据,一经查实,墨天轮将立刻删除相关内容。

评论