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