rocm/migraphx-ci-ubuntu Docker 镜像 - 轩辕镜像 | Docker 镜像高效稳定拉取服务

镜像概述

MIGraphX构建环境预构建Docker镜像，旨在为开发人员提供快速、标准化的环境，用于构建AMD ROCm生态系统中的深度学习推理引擎MIGraphX（GitHub: [***]

核心特性

• 预配置依赖环境：集成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 [***]
cd AMDMIGraphX

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

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