🚀 Ubuntu 26.04 LTS + RTX 5090 完整配置指南
本教程适用于 Ubuntu 26.04 LTS 系统,显卡为 NVIDIA RTX 5090。我们将通过 NVIDIA 官方源安装最新的显卡驱动和 CUDA 工具包,并设置合理的功耗上限(500W)以保障长期稳定运行。所有步骤均适合大模型训练/推理场景。
1. 准备工作
1.1 设置 root 密码(可选)
打开终端(Ctrl+Alt+T),运行:
sudo passwd root
根据提示输入当前用户密码,然后为 root 用户设置一个新密码(输入两次确认)。设置后可使用 su - 切换,但后续命令我们仍推荐使用 sudo。
1.2 更新系统并安装基础编译工具
sudo apt update && sudo apt upgrade -y
sudo apt install -y build-essential linux-headers-$(uname -r) dkms
2. 安装 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 reboot
3. 验证与配置环境变量
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 ~/.bashrc
3.2 验证安装
运行以下命令检查驱动和 CUDA 是否正常工作:
nvidia-smi
nvcc --version
nvidia-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 功耗限制(开机自启)
长期运行大模型时,合理限制功耗上限可以提升稳定性、降低发热。对于 RTX 5090,推荐限制为 500W(低于默认 575W)。
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.service
4.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. 参考资料
评论(0)
暂无评论