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

 

 

我是用的一个笔记本，装的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版本对应图：

下载完成后，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