ROCm在Debian 12系统上的安装与配置指南

背景介绍

AMD ROCm（Radeon Open Compute）平台是AMD推出的开源GPU计算生态系统，专为高性能计算和机器学习任务优化。本文将详细介绍在Debian 12系统上安装和配置ROCm的完整过程，特别针对安装过程中可能遇到的常见问题提供解决方案。

系统要求

在开始安装前，请确保您的系统满足以下要求：

• Debian 12操作系统
• AMD Radeon RX 7900 XTX或兼容的AMD GPU
• 内核版本6.1.0-30或更高
• 已启用Secure Boot（如使用）

安装步骤

1. 准备工作

首先更新系统并安装必要的依赖包：

sudo apt update && sudo apt upgrade -y
sudo apt install linux-headers-$(uname -r) build-essential dkms

2. 安装ROCm

使用AMD提供的官方安装工具进行安装：

sudo apt install amdgpu-install
sudo amdgpu-install --usecase=dkms,rocm

对于需要图形支持的用户，可以使用：

sudo amdgpu-install --usecase=graphics,rocm

3. 处理Secure Boot问题

如果系统启用了Secure Boot，需要为内核模块签名：

1. 生成密钥对：

sudo mkdir -p /var/lib/shim-signed/mok
sudo openssl req -new -x509 -newkey rsa:2048 -keyout /var/lib/shim-signed/mok/MOK.priv -outform DER -out /var/lib/shim-signed/mok/MOK.der -nodes -days 36500 -subj "/CN=Custom MOK/"

2. 导入密钥：

sudo mokutil --import /var/lib/shim-signed/mok/MOK.der

3. 重启系统并按照提示完成密钥注册

4. 配置DKMS自动签名：

echo "mok_signing_key = /var/lib/shim-signed/mok/MOK.priv" | sudo tee -a /etc/dkms/framework.conf
echo "mok_certificate = /var/lib/shim-signed/mok/MOK.der" | sudo tee -a /etc/dkms/framework.conf

4. 验证安装

安装完成后，验证ROCm是否正常工作：

rocminfo

检查GPU状态：

amd-smi

常见问题解决

1. 内核模块加载失败

如果遇到modprobe: ERROR: could not insert 'amdgpu': Key was rejected by service错误，说明Secure Boot阻止了未签名模块的加载。按照上述Secure Boot部分处理。

2. 图形界面卡顿

如果安装后出现图形界面卡顿问题，可以尝试：

• 使用--usecase=dkms,rocm而非图形用例安装
• 确保使用系统提供的Mesa驱动而非AMD提供的版本

3. 环境变量配置

为方便使用，建议设置以下环境变量：

echo 'export ROCM_PATH=/opt/rocm' >> ~/.bashrc
source ~/.bashrc

性能优化建议

1. 对于机器学习工作负载，建议使用--usecase=rocm专用安装
2. 定期检查并更新ROCm版本以获得最佳性能
3. 监控GPU使用情况，确保散热良好

结论

在Debian 12系统上安装ROCm需要特别注意内核模块签名和用例选择。通过正确配置Secure Boot和选择合适的安装用例，可以确保ROCm平台稳定运行，为GPU加速计算任务提供强大支持。本文提供的解决方案经过实际验证，适用于大多数AMD GPU在Debian系统上的部署场景。