安装CUDA

操作步骤

  1. 使用PuTTY工具,以root用户登录服务器。
  2. 执行以下命令查看Nouveau驱动是否已禁用。

    lsmod |grep nouveau
    • 若有回显信息,表示Nouveau驱动未禁用,则执行3
    • 若无回显信息,表示Nouveau驱动已禁用,则结束操作。

  3. 执行以下命令禁用Nouveau驱动。

    1. 新建文件“disable-nouveau.conf”。
      vi /etc/modprobe.d/disable-nouveau.conf
    2. 按“i”进入编辑模式,添加如下内容。
      blacklist nouveau
      options nouveau modeset=0
    3. 按“Esc”键,输入wq!,按“Enter”保存并退出编辑。

  4. 执行以下命令备份并生成新的initramfs文件。

    cp /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r)-nouveau.img 
    dracut -f /boot/initramfs-$(uname -r).img $(uname -r)

  5. 执行以下命令重启机器。

    reboot

  6. 执行以下命令确认Nouveau驱动已禁用(回显为空)。

    lsmod |grep nouveau

  7. 执行以下命令安装NVIDIA CUDA组件。

    wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux_sbsa.run
    sudo sh cuda_11.4.0_470.42.01_linux_sbsa.run

  8. 执行以下命令配置环境变量。

    export CUDA_HOME=/usr/local/cuda
    export PATH=$CUDA_HOME/bin:$PATH
    export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

  9. 执行以下命令验证版本信息。

    nvcc -V

    回显显示如下信息,表示安装成功。