CentOS 7.x 安装CUDA

1. 当前环境

   [root@node17 ~]# cat /etc/redhat-release 
   CentOS Linux release 7.9.2009 (Core
   

2. 首先需要确认自己的GPU是否支持CUDA。查看支持CUDA的GPU列表

   [root@node17 ~]# lspci | grep -i nvidia
   af:00.0 3D controller: NVIDIA Corporation GA100 [GRID A100 PCIe 40GB] (rev a1)
   

3. 验证是否有GCC编译环境。

   [root@node17 ~]# gcc -v
   

4. 验证是否安装了内核头文件和开发包。

   [root@node17 ~]# yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
   

5. 查看本机是否有装nouveau驱动，如果有就禁用掉（因为它和NVIDIA官方的驱动有冲突，所以安装NVIDIA驱动和CUDA前应该禁用它）。如果没有就跳过本步骤。

   [root@node17 ~]# lsmod | grep nouveau    # 如果有结果就执行下面操作禁用掉，没有就忽略。
   [root@node17 ~]# vim /etc/modprobe.d/blacklist-nouveau.conf
   blacklist nouveau
   options nouveau modeset=0
   [root@node17 ~]# mv /boot/initramfs-$(uname -r).img{,.bak}
   [root@node17 ~]# dracut /boot/initramfs-$(uname -r).img $(uname -r)
   ## 重启系统并验证。
   [root@node17 ~]# shutdown -r now
   [root@node17 ~]# lsmod | grep nouveau
   

6. 下载驱动程序，CUDA历史包下载地址。

   [root@node17 ~]# wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-rhel7-11-2-local-11.2.0_460.27.04-1.x86_64.rpm
   [root@node17 ~]# yum -y localinstall cuda-repo-rhel7-11-2-local-11.2.0_460.27.04-1.x86_64.rpm
   [root@node17 ~]# yum -y install nvidia-driver-latest-dkms cuda cuda-drivers
   

7. 验证。

   [root@node17 ~]# cd /usr/local/cuda/samples/1_Utilities/deviceQuery
   [root@node17 ~]# make
   [root@node17 ~]# ./deviceQuery
   

 
 

写作不易，转载请注明出处，谢谢~~