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
上一篇: 天玑1100和骁龙778g有什么区别 天玑1100和骁龙778g性能对比
下一篇: 身后男人的数量