Centos GPU驱动部署(N卡驱动,CUDA,cuDNN)

随笔2个月前发布 卫风
4 0 0

https://blog.csdn.net/bjzyb/article/details/131460278

操作系统:Centos 7
显卡:NVIDIA T4

基础环境准备:
yum -y update
yum -y install kernel-devel
yum -y install epel-release
yum -y install dkms
yum -y install gcc gcc+
查看内核:uname -a

禁用默认显卡:一般情况下公有云云服务器中的GPU机型已经禁用了,检查lsmod | grep nouveau 输出为空即可不用在单独配置;
若自建GPU服务器:
1)安装好依赖包后需要禁用 nouveau,只有在禁用掉 nouveau 后才能顺利安装 NVIDIA 显卡驱动,禁用方法就是在 vim /etc/modprobe.d/blacklist.conf 文件中添加一条禁用命令
blacklist nouveau
options nouveau modeset=0
执行命令:
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

使用 dracut重新建立 initramfs image file :
备份 the initramfs file
$ mv /boot/initramfs-( u n a m e − r ) . i m g / b o o t / i n i t r a m f s − (uname -r).img /boot/initramfs-(uname−r).img/boot/initramfs−(uname -r).img.bak
重新建立 the initramfs file
$ dracut -v /boot/initramfs-$(uname -r).img
3)最后重启电脑
reboot
4)检查nouveau driver确保没有被加载!
$ lsmod | grep nouveau
输出为空即可

安装驱动(410)
在nvidia网站搜索对应版本驱动,下载后传输至服务器
https://www.nvidia.com/Download/index.aspx?lang=en-us

 

安装驱动:
chmod +x NVIDIA-Linux-x86_64-410.129-diagnostic.run
./NVIDIA-Linux-x86_64-410.129-diagnostic.run
注意:由于本次安装410版本,第一步提示,选择NO,安装410版本
后续在安装中会提示32-bit安装,选择No

安装完成后可查看版本: nvidia-smi

部署CUDA(本次版本10.0)
1)在网站选择对应版本:https://developer.nvidia.com/cuda-toolkit-archive
下载完成后传输至服务器,使用rpm安装

 

2)安装cuda-10.0
yum clean all
yum install cuda-10.0.130 –nogpgcheck

查看安装结果: nvcc -V

 

部署cuDNN (本次版本7.4.1)
在网站选择对应版本:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b
下载完成后传输至服务器

修改文件名:
cp cudnn-10.0-linux-x64-v7.4.1.5.tgz cudnn-10.0-linux-x64-v7.4.1.5.tar.gz
解压安装包:
tar zxvf cudnn-10.0-linux-x64-v7.4.1.5.tar.gz
解压后文件夹为:cuda
进入cuda下,copy文件:
cp -P include/cudnn*.h /usr/local/cuda-10.0/include
cp -P lib64/libcudnn* /usr/local/cuda-10.0/lib64

赋予权限:
chmod a+r /usr/local/cuda-10.0/include/cudnn.h
chmod a+r /usr/local/cuda-10.0/lib64/libcudnn*

查看:
cat /usr/local/cuda-10.0/version.txt
cat /usr/local/cuda-10.0/include/cudnn.h | grep CUDNN_MAJOR -A 2

————————————————
版权声明:本文为CSDN博主「bjzyb」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/bjzyb/article/details/131460278

© 版权声明

相关文章

暂无评论

您必须登录才能参与评论!
立即登录
暂无评论...