AMD MI50在Ubuntu 24.04下安装驱动和ROCm完全指南

欢迎来到本教程！如果你拥有一块AMD MI50（Radeon Instinct MI50）加速卡，并希望在最新的Ubuntu 24.04 LTS上充分发挥其计算潜力，那么这篇文章正是为你准备的。我们将一步步带你完成AMD MI50驱动安装和完整的ROCm安装教程，即使你是Linux新手也能轻松跟上。

1. 准备工作：了解你的硬件与系统

AMD MI50基于Vega架构，完全支持ROCm（Radeon Open Compute）平台。在开始之前，请确认你的系统满足以下条件：

• 已安装Ubuntu 24.04 LTS（64位）并更新至最新：sudo apt update && sudo apt upgrade -y
• 拥有root权限或sudo权限
• 确保显卡正确安装，且电源供电充足
• BIOS中开启Above 4G Decoding和Resizable BAR（可选但推荐）

2. 添加ROCm软件源

为了安装最新的驱动和ROCm，我们需要添加AMD的官方仓库。打开终端，执行以下命令：

wget -q -O - https://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -echo "deb [arch=amd64] https://repo.radeon.com/rocm/apt/6.0 noble main" | sudo tee /etc/apt/sources.list.d/rocm.listsudo apt update

注意：这里以ROCm 6.0为例，实际版本请参考官网。添加成功后，系统就能识别到ROCm软件包了。

3. 安装AMDGPU驱动与ROCm

接下来我们一次性安装驱动和核心ROCm包。Ubuntu 24.04的Ubuntu 24.04 ROCm配置过程非常顺畅：

sudo apt install rocm

这个命令会自动安装AMD MI50驱动安装所需的amdgpu-dkms以及ROCm运行时、开发工具等。安装过程中会提示你确认，输入Y即可。整个过程可能需要几分钟，请耐心等待。

4. 用户权限配置

为了让普通用户也能访问GPU，需要将用户添加到render和video组：

sudo usermod -a -G render,video $USER

然后注销并重新登录，或者重启系统使权限生效。

5. 验证安装

现在来检查一切是否正常。运行以下命令：

rocminfoclinfo

如果看到MI50的设备信息，并且没有错误，那么恭喜你，ROCm安装教程部分大功告成！你也可以运行一个简单的测试：

cd /opt/rocm/bin./rocminfo

6. 下一步：使用ROCm加速你的应用

有了ROCm环境，你就可以编译运行支持AMD GPU加速的软件了，比如TensorFlow、PyTorch、OpenMM等。许多HPC和AI框架都通过ROCm提供了对AMD显卡的支持。记得在编译时启用ROCm后端，具体请参考相应文档。

常见问题排查

Q: 安装后系统无法启动？ A: 可能是驱动冲突，尝试在恢复模式下手动安装amdgpu-dkms，或者检查内核版本是否过旧。

Q: rocminfo找不到设备？ A: 确认显卡已正确插入，且电源线连接；尝试重启；检查是否有其他驱动（如NVIDIA）占用。

本教程基于实际安装经验编写，旨在帮助初学者快速上手。如果你在安装过程中遇到任何问题，欢迎在评论区交流讨论！