在 Ubuntu 22.04 上安装 NVIDIA 显卡驱动主要有两种方式：自动安装（推荐） 和 手动指定版本安装。

为了确保你的 Ollama 或其他 AI 程序能顺利调用显卡，建议使用 官方仓库 的驱动，这样最稳定且容易管理。

以下是详细步骤：

第一步：准备工作

  1. 更新系统软件包列表
    打开终端，执行：

    sudo apt update && sudo apt upgrade -y

  2. 确认显卡型号
    确保系统能识别到你的物理显卡：

    ubuntu-drivers devices

    如果你看到类似 model: ... GeForce RTX ... 的输出，说明系统识别到了显卡。

第二步：安装驱动

方法 A：自动安装推荐版本（最简单）

这是最省心的办法，Ubuntu 会自动检测你的显卡并安装最合适的驱动版本（通常是 tested 版）。

sudo ubuntu-drivers autoinstall

方法 B：手动指定版本（更可控）

如果你需要特定的驱动版本（例如为了兼容某些 CUDA 版本，或者自动安装失败），可以手动指定。

  1. 查看推荐的驱动版本：

    ubuntu-drivers devices

    输出示例：

    driver   : nvidia-driver-535 - distro non-free recommended
driver   : nvidia-driver-550 - distro non-free
driver   : nvidia-driver-470 - distro non-free

    可以看到 recommended 标记在哪个版本上（通常选数字最大的稳定版，如 535 或 550）。

  2. 安装指定版本（以 535 为例）：

    sudo apt install nvidia-driver-535 -y

第三步：重启系统（必须）

驱动安装完成后，必须重启计算机才能生效：

sudo reboot

第四步：验证安装

重启回来后，打开终端，输入 NVIDIA 的管理命令：

nvidia-smi

成功的标志：

  1. 你看到了一个表格界面。
  2. 顶部显示了 Driver Version: 535.xx 等信息。
  3. 可以看到显存大小和 GPU 使用率。

关于 CUDA Version 的说明：
在 nvidia-smi 右上角显示的 CUDA Version: 12.x 并不代表你已经安装了 CUDA Toolkit 开发工具包。它只是表示当前驱动最高支持到这个版本的 CUDA。对于运行 Ollama 来说，通常只要驱动装好（能看到这个界面），Ollama 就能自动调用 GPU，不需要额外安装庞大的 CUDA Toolkit。

第五步：常见问题与排错

1. Secure Boot（安全启动）问题 [最常见的大坑]

如果你的主板 BIOS 开启了 Secure Boot，在安装驱动的过程中，终端会弹出一个蓝色背景的界面（MOK Management），要求你设置一个密码。

  • 安装时： 设置一个简单密码（如 12345678）。
  • 重启时： 屏幕会出现蓝色的 Perform MOK management 界面。
    1. 选择 Enroll MOK。
    2. 选择 Continue。
    3. 输入你刚才设置的密码。
    4. 选择 Reboot。
  • 如果不这样做： 驱动虽然安装了，但系统会因为签名验证失败而拒绝加载驱动，导致 nvidia-smi 报错。
  • 简单粗暴的解法： 直接进入 BIOS 关闭 Secure Boot。

2. 安装失败或冲突

如果你之前乱装过驱动，或者安装了一半失败了，建议先清理干净再重装：

# 卸载所有 nvidia 相关包
sudo apt purge nvidia* libnvidia* -y
sudo apt autoremove -y
# 重启
sudo reboot
# 然后重新按照上面的方法 A 安装

针对 Ollama 用户的补充

一旦 nvidia-smi 能正常显示，你不需要做任何额外配置。重启 Ollama 服务，它会自动检测到 GPU。

你可以通过查看 Ollama 的运行日志来确认它是否使用了 GPU：

journalctl -u ollama -n 50

如果在日志中看到类似 GPU detected 或 compute capability 的字样，说明你的显卡已经在为 AI 加速了。