前言:
做人脸识别比对和检索服务,服务器用CPU进行人脸识别速度慢,用GPU来进行加速计算
环境:
CentOS 7.9
显卡型号:
技嘉RTX2060-6G
注:Ubuntu的用户可以很方便的进行驱动安装,并不需要手动去安装折腾
首先下载英伟达显卡驱动
官网地址:NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA
下载选项大概就是这样子(根据你自己的显卡型号和对应的操作系统)
下载下来的名字类似 NVIDIA-Linux-x86_64-465.24.02.run 这样子 以.run结尾的文件
步骤一(可以升级内核或者跳过直接看步骤二):
要装的三个依赖分别是,gcc、kernel-devel、dkms,其中需要注意的是,kernel-devel的版本需要与当前内核的版本一致,不然后面会出现找不到文件的情况。
查看自己的内核版本
[root@localhost ~]# uname -r
3.10.0-1160.el7.x86_64
3.10.0-1160.el7.x86_64就是我的内核版本号
然后用yum list 去查看我可以安装的依赖
[root@localhost ~]# yum list | grep kernel-devel
kernel-devel.x86_64 3.10.0-1160.24.1.el7 @updates
查出来发现对应的版本比我查出来的内核多了个 24.1
在系统里其实内核已经更新到最新了,也就是3.10.0-1160.24.1.el7
可以看看这个目录是否存在
cd /usr/src/kernels/3.10.0-1160.24.1.el7.x86_64
如果确实需要完全对应版本的 kernel-devel-3.10.0-1160.24.1.el7.rpm 包
可以点这里下载,免费提供的,如果下载需要C币可以留邮箱我直接发给你(C币这个东西真不是我设置的)
kernel-devel-3.10.0-1160.el7.x86_64.rpm
步骤二:
安装其他依赖
yum -y install gcc dkms
yum install kernel-devel kernel-doc kernel-headers gcc\* glibc\* glibc-\*
默认情况下,nouveau是启动的,这个和安装NVIDIA驱动是有冲突的,所以要阻止 nouveau 模块的加载
修改位于/etc/modprobe.d/blacklist.conf 文件:
如果存在blacklist.conf,添加blacklist nouveau,注释掉blacklist nvidiafb;
如果不存在,执行:
echo -e "blacklist nouveau\noptions nouveau modeset=0" > /etc/modprobe.d/blacklist.conf
重新建立initramfs image文件
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
dracut /boot/initramfs-$(uname -r).img $(uname -r)
这个过程可能稍久一点,不要急,等待建立完毕
把下载下来的驱动文件给下权限
chmod u+x NVIDIA-Linux-x86_64-465.24.02.run
安装驱动(这里明确指向了内核的路径)
./NVIDIA-Linux-x86_64-465.24.02.run --kernel-source-path=/usr/src/kernels/3.10.0-1160.24.1.el7.x86_64 -k $(uname -r)
如果出现以下报错
则需要进入命令行模式,退出GUI界面
init 3
然后再执行安装驱动那一步
接着会出现这个界面
选择continue installation
选择no
选择Yes
成功安装
主要按照提示一步步安装就好了
最后输入
nvidia-smi
验证一下
出现以上界面的话就说明安装成功了