开发者
我要评分
获取效率
正确性
完整性
易理解
在线提单
论坛求助

部署CUDA

  1. 使用PuTTY工具,以root用户登录服务器。
  2. 查看nouveau驱动是否已禁用。
    lsmod |grep nouveau
    • 回显不为空表示未禁用,执行3~6操作禁用nouveau驱动。
    • 回显为空表示已禁用,跳转到7执行后续操作。
  3. 禁用nouveau驱动。
    1. 新建文件“/etc/modprobe.d/disable-nouveau.conf”。
      vim /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

    回显显示版本信息即为安装成功。