欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

centos7.6安装NVIDIA驱动、CUDA、cuDNN成功后的个人总结

程序员文章站 2022-03-06 14:29:51
...

 

 

我是用的一个笔记本,装的windows+centos双系统

系统版本:centos7.6(64位)

显卡:GTX1070 mobile

处理器:i7-6700k

内存:16G

一、安装NVIDIA驱动

1、安装编译环境:gcc、kernel-devel、kernel-headers("kernel-devel-uname-r == $(uname -r)"可以确保安装与当前运行内核版本一样的kernel-header)

yum -y install gcc kernel-devel "kernel-devel-uname-r == $(uname -r)" dkms

2、修改/etc/modprobe.d/blacklist.conf 文件,以阻止 nouveau 模块的加载(若没有vim命令,运行yum install vim来安装vim)

vim /etc/modprobe.d/blacklist.conf

 添加:blacklist nouveau,注释掉blacklist nvidiafb(如果存在)

3、重新建立initramfs image文件(生成新的内核,这个内核在开机的时候不会加载nouveau驱动程序)

mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
dracut /boot/initramfs-$(uname -r).img $(uname -r)
reboot

4、重启后输入:lsmod | grep nouveau,没有任何输出,则确定nouveau没有加载

5、前往https://www.nvidia.cn/object/unix-cn.html找到自己需要的NVIDIA驱动,因本人需要装cuda10.1,所以安装了418.88的驱动版本,附上NVIDIA驱动与cuda版本对应图:

centos7.6安装NVIDIA驱动、CUDA、cuDNN成功后的个人总结

下载完成后,ssh连接centos上传至home下

cd /home
chmod 777 NVIDIA-Linux-x86_64-418.88.run
./NVIDIA-Linux-x86_64-418.88.run --kernel-source-path=/usr/src/kernels/3.10.0-1062.7.1.el7.x86_64/ --no-drm

(注意:--no-drm要带上,要不然安装过程会报错ERROR: The nvidia-drm kernel module failed to load. This kernel module isrequired for the proper operation of DRM-KMS. If you do not need touse DRM-KMS, you can try to install this driver package again withthe '--no-drm' option.)

安装完成后,重启

reboot

6、重启后,输入nvidia-smi,出现显卡配置信息,说明NVIDIA驱动安装成功

二、安装CUDA

1、前往https://developer.nvidia.com/cuda-toolkit-archive选择版本进行下载,下载完成后,上传至home下

cd /home
chmod 777 cuda_10.1.105_418.39_linux.run
./cuda_10.1.105_418.39_linux.run --kernel-source-path=/usr/src/kernels/3.10.0-1062.7.1.el7.x86_64/

(注:--kernel-source-path=/usr/src/kernels/3.10.0-1062.7.1.el7.x86_64/参数要带上,不然会报错,后面的内核版本改成自己的)

选择accept,安装时去掉Driver选项,因为刚才已经单独安装过了。然后install,等待完成。

2、配置环境变量:

vim /etc/profile

在最后添加:

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

3、保存退出,立即生效:

source /etc/profile

4、重启:

reboot

5、看版本,验证CUDA安装完成:

nvcc -V

三、安装cuDNN

1、前往https://developer.nvidia.com/rdp/cudnn-archive下载对应的版本后,上传至home下

cd /home
chmod 777 cudnn-10.1-linux-x64-v7.6.4.38.tgz.tgz
tar -zxvf cudnn-10.1-linux-x64-v7.6.4.38.tgz.tgz
cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
chmod a+r /usr/local/cuda-10.1/include/cudnn.h /usr/local/cuda-10.1/lib64/libcudnn*

2、测试:

cd /root/NVIDIA_CUDA-10.1_Samples
cd 1_Utilities/deviceQuery
make
./deviceQuery

 

相关标签: AI