ROCm在Ubuntu 24.04上的安装问题分析与解决方案

问题背景

AMD的ROCm（Radeon Open Compute）平台是面向高性能计算和机器学习开发的开放软件生态系统。近期有用户在Ubuntu 24.04（代号Noble）系统上安装ROCm 6.3.3版本时遇到了依赖关系问题，导致安装失败。

问题现象

用户在尝试安装ROCm 6.3.3时，系统报告了多个依赖包无法满足的问题，主要包括：

1. libzstd-dev包依赖的libzstd1版本不匹配
2. mesa-common-dev包缺少libdrm-dev依赖
3. python3-dev包依赖的python3版本不匹配
4. zlib1g-dev包依赖的zlib1g版本不匹配

这些错误提示表明系统无法正确解析和安装所需的依赖关系。

问题根源分析

经过技术分析，发现问题的根本原因在于Ubuntu 24.04的软件源配置不完整。Ubuntu系统通常需要配置多个软件源套件（suites）才能获取所有必要的软件包更新：

• noble：基础软件包
• noble-updates：重要更新包
• noble-backports：向后移植的新功能
• noble-security：安全更新

用户初始配置中缺少了noble-updates和noble-backports这两个关键套件，导致系统无法获取最新版本的依赖包。

解决方案

要解决这个问题，需要完整配置Ubuntu 24.04的软件源。以下是具体步骤：

1. 首先清理现有的ROCm安装残留（如果存在）：

sudo amdgpu-install --uninstall --rocmrelease=all
sudo apt purge amdgpu-install
sudo apt autoremove

2. 更新软件源配置文件/etc/apt/sources.list.d/ubuntu.sources，确保包含所有必要的套件：

sudo tee /etc/apt/sources.list.d/ubuntu.sources > /dev/null <<EOF
Types: deb
URIs: http://archive.ubuntu.com/ubuntu/
Suites: noble noble-updates noble-backports
Components: main restricted universe multiverse
Signed-By: /usr/share/keyrings/ubuntu-archive-keyring.gpg

Types: deb
URIs: http://security.ubuntu.com/ubuntu/
Suites: noble-security
Components: main restricted universe multiverse
Signed-By: /usr/share/keyrings/ubuntu-archive-keyring.gpg
EOF

3. 更新软件包列表并安装必要的系统组件：

sudo apt update
sudo apt install "linux-headers-$(uname -r)" "linux-modules-extra-$(uname -r)"
sudo apt install python3-setuptools python3-wheel

4. 将当前用户添加到必要的用户组：

sudo usermod -a -G render,video $LOGNAME

5. 安装ROCm 6.3.3：

wget https://repo.radeon.com/amdgpu-install/6.3.3/ubuntu/noble/amdgpu-install_6.3.60303-1_all.deb
sudo apt install ./amdgpu-install_6.3.60303-1_all.deb
sudo apt update
sudo apt install amdgpu-dkms rocm

技术要点

1. 软件源配置的重要性：Ubuntu系统依赖正确的软件源配置来解析和安装软件包。不完整的配置会导致依赖关系无法满足。

2. 用户组权限：将用户添加到render和video组是必要的，这确保了用户有权限访问GPU设备。

3. 内核模块支持：安装与当前运行内核匹配的linux-headers和linux-modules-extra包，确保DKMS（动态内核模块支持）能够正确构建和安装AMDGPU驱动模块。

最佳实践建议

1. 在安装ROCm前，始终检查并确保系统软件源配置完整。

2. 考虑使用最新稳定版的ROCm（当前为6.4.0），除非有特定需求需要使用旧版本。

3. 安装完成后，建议运行简单的测试命令（如rocminfo）验证安装是否成功。

4. 对于生产环境，建议在安装前创建系统快照或备份，以便在出现问题时能够快速恢复。

通过遵循上述解决方案和最佳实践，用户应该能够在Ubuntu 24.04系统上顺利安装和使用ROCm平台。