安装NVIDIA CUDA组件

操作步骤

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

   lsmod |grep nouveau

   • 若有回显信息，表示Nouveau驱动未禁用，则执行3。
   • 若无回显信息，表示Nouveau驱动已禁用，则执行4。

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. 执行以下命令安装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 shcuda_11.4.0_470.42.01_linux_sbsa.run

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

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

6. 执行以下命令进行验证。

   nvcc --version

   若返回结果已包含NVIDIA CUDA组件版本信息，说明安装成功。