AMD MI50 显卡在 Ubuntu 24.04 上的全攻略（从驱动安装到 ROCm 环境搭建）

AMD MI50 显卡在 Ubuntu 24.04 上的全攻略（从驱动安装到 ROCm 环境搭建）

本文关键词：AMD MI50, Ubuntu 24.04, ROCm 教程, 显卡驱动安装

一、前言

AMD Radeon Instinct MI50 是一款强大的计算加速卡，广泛应用于深度学习和高性能计算（HPC）。随着 Ubuntu 24.04 LTS（Noble Numbat）的发布，很多开发者希望在这个最新的系统上运行 ROCm。本文将手把手教你如何完成驱动与 ROCm 环境的部署，即使是 Linux 小白也能轻松上手。

二、准备工作

在开始安装前，请确保你的系统已更新，并且拥有 sudo 权限。首先，打开终端并执行以下命令：

sudo apt update && sudo apt upgrade -ysudo apt install wget gnupg2 shell-scripts -y

三、添加 AMD ROCm 仓库

为了获取针对 Ubuntu 24.04 优化的驱动和软件包，我们需要手动添加 AMD 的官方软件源。请依次输入以下指令：

# 下载并添加 GPG 密钥sudo mkdir -p /etc/apt/keyringswget -qO - https://repo.radeon.com/rocm/rocm.gpg.key | sudo gpg --dearmor -o /etc/apt/keyrings/rocm.gpg# 添加 ROCm 6.1 (或最新版本) 软件源echo "deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/6.1 noble main" | sudo tee /etc/apt/sources.list.d/rocm.list

注意：这里的 "noble" 对应 Ubuntu 24.04 的版本代号。

四、安装内核驱动与 ROCm

这是显卡驱动安装中最关键的一步。MI50 需要特定的内核组件才能发挥性能。

sudo apt updatesudo apt install amdgpu-dkms rocm-hip-sdk -y

安装过程可能需要几分钟，具体取决于你的网络速度。安装完成后，请务必将当前用户添加到 render 和 video 组，否则你将没有权限调用 GPU：

sudo usermod -aG render $USERsudo usermod -aG video $USER

提示：执行完上述命令后，请务必重启电脑以使配置生效。

五、验证安装是否成功

重启后，我们可以通过 ROCm 自带的工具来检测 AMD MI50 是否被正确识别。在终端输入：

/opt/rocm/bin/rocminfo

如果输出信息中包含了 "Device Name: gfx906"（MI50 的核心代号），那么恭喜你，驱动和 ROCm 环境已经安装成功了！

六、总结

通过这篇 ROCm 教程，我们完成了在 Ubuntu 24.04 上为 AMD MI50 配置开发环境的所有步骤。现在，你可以开始尝试运行 PyTorch 或 TensorFlow 的 ROCm 版本，开启你的 AI 算力之旅了！