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
    


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

posted @ 2021-09-23 10:49  merlee  阅读(1709)  评论(0编辑  收藏  举报