rocm/migraphx-ci-sles Docker 镜像 - 轩辕镜像

镜像概述

本镜像提供基于SLES（SUSE Linux Enterprise Server）的MIGraphX预构建环境，旨在简化AMD MIGraphX机器学习推理引擎的部署与使用。MIGraphX是AMD开发的开源机器学习推理引擎，支持深度学习模型的优化、编译和高效推理，适用于各类深度学习应用场景。通过本预构建镜像，用户无需手动编译和配置MIGraphX及其依赖，可直接启动环境进行模型推理和应用开发。

核心功能与特性

• 预配置MIGraphX环境：镜像已内置编译好的MIGraphX库及相关依赖（如ROCm运行时、深度学习框架支持库），无需用户手动编译，开箱即可使用。
• SLES系统基础：基于企业级SLES系统构建，提供稳定、安全的运行环境，适合生产环境部署。
• 与AMDMIGraphX项目同步：镜像构建与AMDMIGraphX GitHub项目保持版本同步，确保集成最新功能与修复。
• 完整工具链支持：包含MIGraphX命令行工具（如migraphx-driver模型转换工具、migraphx-opt模型优化工具）及示例代码，便于快速验证和开发。

使用场景与适用范围

• 企业级推理部署：在SLES环境中快速部署基于MIGraphX的机器学习推理服务，适用于***、制造等对系统稳定性要求高的行业。
• 深度学习应用开发：基于SLES系统开发依赖MIGraphX的应用时，无需手动配置底层环境，加速开发流程。
• 快速验证场景：需要临时或快速启动MIGraphX环境验证模型性能、测试兼容性的场景（如学术研究、原型验证）。

使用方法与配置说明

前提条件

• 已安装Docker引擎（推荐版本20.10及以上）。
• 若需启用GPU加速，主机需配置ROCm驱动（MIGraphX依赖ROCm实现GPU加速推理）。

获取镜像

通过官方仓库拉取镜像（具体仓库地址请参考AMDMIGraphX项目文档）：

bash
docker pull <镜像仓库地址>/sles-migraphx:<标签>
# 示例（假设使用latest标签）：
# docker pull rocmsoftwareplatform/sles-migraphx:latest

基本运行命令

启动交互式容器（适用于环境验证和手动操作）：

bash
docker run -it --rm <镜像名称>:<标签> /bin/bash

GPU加速配置

若主机已安装ROCm驱动，通过以下命令启用GPU支持：

bash
docker run -it --rm \
  --device=/dev/kfd \          # 访问ROCm设备
  --device=/dev/dri \          # 访问GPU显示设备
  --group-add video \          # 添加视频组权限
  <镜像名称>:<标签> /bin/bash

环境验证

容器启动后，通过以下命令验证MIGraphX是否正常工作：

bash
# 查看MIGraphX版本
migraphx-driver --version

# 运行示例模型推理（若镜像包含示例）
migraphx-driver run --model /path/to/example/model

高级配置与扩展

镜像的详细配置选项（如环境变量、模型挂载路径）、自定义编译参数及扩展功能，请参考AMDMIGraphX官方GitHub仓库的文档与示例。