rocm/migraphx-ci-ubuntu

镜像概述

MIGraphX构建环境预构建Docker镜像，旨在为开发人员提供快速、标准化的环境，用于构建AMD ROCm生态系统中的深度学习推理引擎MIGraphX（GitHub: https://github.com/ROCm/AMDMIGraphX%EF%BC%89%E3%80%82%E9%95%9C%E5%83%8F%E9%A2%84%E5%85%88%E9%9B%86%E6%88%90%E6%9E%84%E5%BB%BA%E6%89%80%E9%9C%80%E7%9A%84%E5%85%A8%E9%83%A8%E4%BE%9D%E8%B5%96%E7%BB%84%E4%BB%B6%EF%BC%8C%E7%AE%80%E5%8C%96%E4%BB%8E%E6%BA%90%E7%A0%81%E6%9E%84%E5%BB%BAMIGraphX%E7%9A%84%E6%B5%81%E7%A8%8B%E3%80%82

核心特性

• 预配置依赖环境：集成MIGraphX构建所需的编译器、ROCm库、深度学习框架依赖及工具链，避免手动安装复杂依赖。
• ROCm版本兼容：与指定版本的ROCm生态系统深度兼容，确保构建产物可无缝对接AMD GPU硬件。
• 标准化构建流程：提供一致的构建环境，消除本地环境差异导致的构建问题，提升团队协作效率。

使用场景

• MIGraphX开发：开发人员快速搭建本地构建环境，专注于功能开发与代码优化。
• CI/CD集成：作为持续集成/部署流程的基础镜像，自动化执行MIGraphX的构建、测试与版本发布。
• 快速环境验证：科研人员或用户快速构建特定版本的MIGraphX，验证新特性、修复或性能优化。

Docker部署示例

1. 拉取镜像

bash
docker pull <image-name>  # 替换为实际镜像名称（如rocm/migraphx-build-env）

2. 启动容器

bash
docker run -it --rm --privileged <image-name> /bin/bash

说明：--privileged 选项用于支持GPU访问（如适用）；--rm 确保容器退出后自动清理。

3. 构建MIGraphX

在容器内执行以下命令构建源码：

bash
# 克隆MIGraphX源码（如镜像未预包含）
git clone https://github.com/ROCm/AMDMIGraphX.git
cd AMDMIGraphX

# 配置构建选项
mkdir build && cd build
cmake .. -DROCM_PATH=/opt/rocm  # 需匹配容器内ROCm安装路径

# 编译源码
make -j$(nproc)  # 使用所有可用CPU核心加速构建