本教程适用于 Ubuntu 26.04 LTS 系统，显卡为 NVIDIA RTX 5090。我们将通过 NVIDIA 官方源安装最新的显卡驱动和 CUDA 工具包，并（可选）设置功耗上限（500W）以提升长期运行稳定性。所有步骤均适合大模型训练/推理场景。
1. 准备工作
1.1 关闭 BIOS 安全启动（Secure Boot）
NVIDIA 开源或闭源驱动在启用安全启动的 UEFI 系统上加载时，需要签名或手动操作，这可能导致驱动无法正常加载。为了简化安装流程，建议先关闭 BIOS 中的 Secure Boot：
- 重启电脑，开机时按 Del、F2、Esc 或对应键进入 BIOS 设置。
- 找到 Security 或 Boot 选项卡，将 Secure Boot 设置为 Disabled。
- 保存设置并退出（通常按 F10）。
⚠️ 关闭安全启动会降低系统安全性，请确保你了解其影响。如果你需要保持安全启动，则需额外为驱动签名，具体步骤可参考 NVIDIA 官方文档。
1.2 设置 root 密码（可选）
打开终端（Ctrl+Alt+T），运行：
sudo passwd root根据提示输入当前用户密码，然后为 root 用户设置一个新密码（输入两次确认）。设置后可使用 su - 切换，但后续命令我们仍推荐使用 sudo。
1.3 更新系统并安装基础编译工具
sudo apt update && sudo apt upgrade -y
sudo apt install -y build-essential linux-headers-$(uname -r) dkms2. 安装 NVIDIA 驱动与 CUDA
2.1 添加 NVIDIA 官方 CUDA 软件源（适用于 Ubuntu 26.04）
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2604/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt update
✅ 此步骤将官方源加入系统，确保后续安装的驱动和 CUDA 工具包来自 NVIDIA。
2.2 安装开源驱动（推荐）
对于 RTX 5090（Blackwell 架构），NVIDIA 官方推荐使用 开源内核模块：
sudo apt install -y nvidia-open
📌 如果你更倾向闭源驱动，可替换为
sudo apt install -y cuda-drivers。
2.3 安装完整的 CUDA 工具包
sudo apt install -y cuda-toolkit该命令会安装 NVIDIA 源中提供的最新稳定版本。
2.4 第一次重启
驱动安装完成后，必须重启系统以加载新驱动：
sudo reboot3. 验证与配置环境变量
3.1 配置 CUDA 环境变量（重启后执行）
重新登录，编辑 ~/.bashrc 文件并添加以下内容：
cat >> ~/.bashrc << 'EOF'
export CUDA_HOME=/usr/local/cuda
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
EOF
source ~/.bashrc3.2 验证安装
运行以下命令检查驱动和 CUDA 是否正常工作：
nvidia-smi
nvcc --versionnvidia-smi应显示驱动版本（如 595.xx）和 GPU 型号 RTX 5090。nvcc --version应显示 CUDA 版本（如 13.x）。
3.3 安装 nvtop 监控工具（可选但推荐）
sudo add-apt-repository multiverse -y
sudo apt update
sudo apt install -y nvtop运行 nvtop 即可实时查看 GPU 功耗、温度、显存占用等信息。
4. （附加）配置 RTX 5090 功耗限制（在不稳定情况下使用）
此部分为 可选附加步骤，仅在您遇到系统不稳定、过热或电源供应不足时建议执行。通过降低功耗上限（例如设为 500W）可以提升长期运行稳定性，但会轻微影响峰值性能。如果系统运行正常，可跳过此部分。
4.1 查看显卡支持的功耗范围
sudo nvidia-smi -q -d POWER | grep -E "Power Limit|Min|Max"确认 500W 在 Min 和 Max 之间（通常支持）。
4.2 创建 systemd 服务（开机自动设置）
创建服务文件：
sudo nano /etc/systemd/system/nvidia-power-limit.service粘贴以下内容：
[Unit]
Description=NVIDIA GPU Power Limit Configuration
After=network.target
[Service]
Type=oneshot
ExecStart=/usr/bin/nvidia-smi -pm 1
ExecStart=/usr/bin/nvidia-smi -i 0 -pl 500
RemainAfterExit=yes
[Install]
WantedBy=multi-user.target保存（Ctrl+O，回车）并退出（Ctrl+X）。
4.3 启用并启动服务
sudo systemctl daemon-reload
sudo systemctl enable nvidia-power-limit.service
sudo systemctl start nvidia-power-limit.service4.4 第二次重启，验证自动生效
sudo reboot重启后，运行以下命令检查功耗限制是否自动应用：
sudo systemctl status nvidia-power-limit.service
sudo nvidia-smi -q -d POWER | grep "Current Power Limit"如果显示 Current Power Limit : 500.00 W，则表示配置成功。
5. 日常使用说明
- 如果您启用了功耗限制服务，每次重启后它会自动运行，无需手动操作。
- 如果需要临时修改功耗值，可编辑服务文件中的
-pl参数，然后执行：sudo systemctl daemon-reload sudo systemctl restart nvidia-power-limit.service - 检查驱动或 CUDA 版本：
nvidia-smi和nvcc --version。
6. 常见问题
❓ 为什么使用
nvidia-open 而不是官网的 .run 文件？nvidia-open 是 NVIDIA 官方发布的开源内核模块驱动，通过 APT 安装与系统集成度更好，更新卸载方便，且对 RTX 5090 等新显卡支持更佳。官网的 .run 文件虽然也可用，但容易造成系统混乱，不推荐新手使用。
❓ 如果
确保已正确添加 NVIDIA 源，然后尝试
cuda-toolkit 找不到怎么办？确保已正确添加 NVIDIA 源，然后尝试
sudo apt update。如果仍找不到，可改用元包 sudo apt install cuda。
❓ 功耗限制设为 500W 会影响性能吗？
对于大模型推理，500W 已足够满足绝大多数场景，且能显著降低温度和功耗，提升长期运行稳定性。若您追求极致性能，可保持默认功耗上限。
对于大模型推理，500W 已足够满足绝大多数场景，且能显著降低温度和功耗，提升长期运行稳定性。若您追求极致性能，可保持默认功耗上限。
7. 参考资料
🎉 恭喜！你的 Ubuntu 26.04 + RTX 5090 环境已完全配置完毕，可以开始大模型开发了！
评论（0）
暂无评论