安装NVIDIA-CUDA组件(含GPU驱动)

操作步骤

  1. 使用PuTTY工具,以root用户登录服务器。
  2. 执行以下命令确认nouveau驱动已禁用(回显为空)。

    lsmod |grep nouveau

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

    wget https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda_11.4.1_470.57.02_linux_sbsa.run
    sh cuda_11.4.1_470.57.02_linux_sbsa.run

    根据提示安装所有组件。

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

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

  5. 执行以下命令查看nvcc版本。

    nvcc -V