鲲鹏社区首页
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

安装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

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