AMD ROCm开源计算平台部署指南

1. 什么是ROCm计算平台？

ROCm（Radeon Open Compute）是AMD开发的开源异构计算平台，提供从底层驱动到应用程序接口的完整软件栈。该平台基于HIP（异构计算接口）构建，支持跨架构的GPU编程，广泛应用于高性能计算（HPC）、人工智能训练推理、科学计算等场景。其核心优势在于开放生态和多架构支持，允许开发者在AMD GPU上实现高效的并行计算。

2. 哪些硬件支持ROCm部署？

ROCm平台对硬件有特定要求，目前支持以下配置：

• GPU型号：AMD Radeon Instinct系列、Radeon Pro系列及Radeon VII等专业显卡
• 架构支持：基于GCN（Graphics Core Next）和RDNA架构的GPU
• 系统架构：x86_64和ARM64架构的处理器

3. 如何准备操作系统环境？

3.1 支持的操作系统版本

ROCm当前支持以下Linux发行版：

• Ubuntu 20.04 LTS
• Ubuntu 22.04 LTS
• Ubuntu 24.04 LTS
• RHEL/CentOS 8.x
• SLES 15

3.2 必备工具安装

🔧 安装版本控制工具

# 安装Git和Git LFS（大文件支持）
sudo apt-get update && sudo apt-get install -y git git-lfs

🔧 配置Git LFS

# 初始化Git LFS支持
git lfs install

🔧 安装Repo多仓库管理工具

# 创建工具目录并下载repo工具
mkdir -p ~/bin
curl https://storage.googleapis.com/git-repo-downloads/repo > ~/bin/repo
chmod a+x ~/bin/repo

# 将工具目录添加到环境变量
echo 'export PATH="$HOME/bin:$PATH"' >> ~/.bashrc
source ~/.bashrc

4. 如何获取ROCm源代码？

📋 步骤1：创建工作目录

# 创建并进入ROCm项目目录
mkdir -p ~/projects/ROCm
cd ~/projects/ROCm

📋 步骤2：初始化代码仓库

# 初始化repo仓库，指定分支为rocm-6.2.x
repo init -u https://gitcode.com/GitHub_Trending/ro/ROCm -b roc-6.2.x

📋 步骤3：同步源代码

# 同步所有组件代码（此过程可能需要较长时间）
repo sync -j$(nproc)

5. 如何配置编译环境？

5.1 安装构建依赖

# 安装基础编译工具
sudo apt-get install -y build-essential cmake ninja-build

# 安装ROCm构建依赖
sudo apt-get install -y libnuma-dev libelf-dev libdrm-dev

5.2 设置环境变量

# 配置ROCm版本
export ROCM_VERSION=6.2.2

# 设置GPU架构目标（根据实际硬件调整）
export GPU_ARCHS="gfx940 gfx941 gfx942"

# 设置并行编译任务数
export NPROC=$(nproc)

6. 如何编译与安装ROCm？

6.1 配置构建系统

# 创建构建目录
mkdir -p ~/projects/ROCm/build
cd ~/projects/ROCm/build

# 生成构建配置
cmake -DCMAKE_INSTALL_PREFIX=/opt/rocm-${ROCM_VERSION} \
      -DROCM_VERSION=${ROCM_VERSION} \
      -DGPU_TARGETS=${GPU_ARCHS} \
      ..

6.2 执行编译过程

# 开始并行编译
make -j${NPROC}

6.3 安装ROCm平台

# 安装到系统目录
sudo make install

# 创建符号链接以便版本管理
sudo ln -s /opt/rocm-${ROCM_VERSION} /opt/rocm

6.4 配置环境变量

# 将ROCm添加到系统路径
echo 'export PATH="/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin:$PATH"' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH="/opt/rocm/lib:/opt/rocm/lib64:$LD_LIBRARY_PATH"' >> ~/.bashrc

# 应用环境变量
source ~/.bashrc

7. 常用构建工具与脚本说明

7.1 ROCm构建配置文件

ROCm/tools/rocm-build/ROCm.mk

该Makefile提供统一的构建入口，支持组件选择性编译和整体构建，通过make rocm-dev可构建完整开发环境。

7.2 依赖安装脚本

ROCm/tools/rocm-build/docker/ubuntu20/install-prerequisites.sh

自动化安装脚本，可根据目标系统自动配置编译依赖，支持Ubuntu、RHEL等多个发行版。

8. 常见问题排查

8.1 编译错误："GPU architecture not supported"

问题描述：编译过程中提示目标GPU架构不支持
解决方法：

1. 确认GPU型号对应的架构代码（可通过rocminfo命令查询）
2. 更新GPU_ARCHS环境变量，仅包含支持的架构

   export GPU_ARCHS="gfx906"  # 替换为实际支持的架构代码
   

8.2 运行时错误："library not found"

问题描述：执行ROCm程序时提示共享库缺失
解决方法：

1. 确认LD_LIBRARY_PATH包含ROCm库路径

   echo $LD_LIBRARY_PATH | grep /opt/rocm/lib
   

2. 若未包含，重新配置环境变量并生效

   source ~/.bashrc
   

8.3 repo同步失败

问题描述：执行repo sync时出现网络错误或认证失败
解决方法：

1. 检查网络连接状态
2. 确认Git配置正确

   git config --global user.email "[e-mail]"
   git config --global user.name "Your Name"
   

3. 清理repo缓存后重试

   repo forall -c "git reset --hard"
   repo sync -j$(nproc)
   

9. 验证安装结果

安装完成后，可通过以下命令验证ROCm环境：

# 查看ROCm版本信息
rocminfo | grep "ROCm Version"

# 运行带宽测试工具
rocm-bandwidth-test

# 执行示例程序
cd /opt/rocm/share/rocblas/examples
make
./rocblas-example-sscal

通过以上步骤，您已完成ROCm开源计算平台的部署。该平台为AMD GPU提供完整的软件支持，可满足从基础科学计算到人工智能训练的多样化需求。