当前位置:首页 > 系统教程 > 正文

Ubuntu系统NVIDIA显卡驱动安装全攻略(CUDA配置与优化详细教程)

Ubuntu系统NVIDIA显卡驱动安装全攻略(CUDA配置与优化详细教程)

Ubuntu系统上安装NVIDIA显卡驱动和CUDA是进行深度学习、科学计算或游戏加速的关键步骤。本教程专为小白设计,一步步指导你完成安装,避免常见陷阱。

Ubuntu系统NVIDIA显卡驱动安装全攻略(CUDA配置与优化详细教程) Ubuntu  NVIDIA驱动 CUDA安装 显卡驱动 第1张

1. 安装前准备

首先,确认你的Ubuntu系统版本和NVIDIA显卡型号。打开终端(Ctrl+Alt+T),输入:

    lspci | grep -i nvidia  

如果输出显示NVIDIA显卡信息,则准备继续。同时,更新系统包列表:

    sudo apt update && sudo apt upgrade -y  

2. 安装NVIDIA显卡驱动

安装NVIDIA驱动有多种方法,这里推荐使用Ubuntu官方仓库自动安装。在终端中运行:

    sudo ubuntu-drivers autoinstall  

或者,手动选择驱动版本。首先查看可用驱动:

    ubuntu-drivers devices  

然后安装推荐驱动(例如nvidia-driver-535):

    sudo apt install nvidia-driver-535 -y  

安装完成后,重启系统:

    sudo reboot  

重启后,验证显卡驱动是否成功:

    nvidia-smi  

如果显示GPU信息,则驱动安装成功。

3. 安装CUDA工具包

CUDA安装需从NVIDIA官网下载。访问 CUDA下载页面,选择Linux、x86_64、Ubuntu和你需要的版本。例如,下载CUDA 11.8安装脚本:

    wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run  

运行安装脚本:

    sudo sh cuda_11.8.0_520.61.05_linux.run  

在安装界面中,接受许可协议,取消勾选驱动安装(因驱动已装),然后安装。完成后,配置环境变量。编辑~/.bashrc文件:

    nano ~/.bashrc  

在文件末尾添加:

    export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}  

保存退出(Ctrl+X,然后Y,回车),然后应用配置:

    source ~/.bashrc  

4. 验证安装结果

检查NVIDIA驱动CUDA安装是否正常工作。运行:

    nvcc -V  

输出CUDA版本信息,则表示CUDA安装成功。同时,再次运行nvidia-smi确认驱动运行。

5. 常见问题解决

  • 黑屏或登录循环:可能是驱动冲突。尝试在恢复模式中卸载驱动:
    sudo apt purge nvidia-* -y
    然后重新安装。
  • CUDA路径错误:确保环境变量设置正确,检查CUDA安装路径。
  • Ubuntu内核更新导致驱动失效:重新安装驱动或使用DKMS模块。

本教程涵盖了Ubuntu系统上NVIDIA驱动CUDA安装的核心步骤。遵循这些步骤,你应该能顺利完成显卡驱动和CUDA配置。如有问题,参考NVIDIA官方文档或社区论坛。