rocm/dev-ubuntu-20.04

ROCM基础开发镜像

镜像概述

本镜像基于Ubuntu 20.04构建，预安装来自repo.radeon.com的rocm-dev开发包，不含内核级组件。主要用途是作为基础镜像，供下游镜像通过Dockerfile的FROM语句派生，构建基于ROCM的应用或开发环境。

核心功能与特性

• 基础系统：基于Ubuntu 20.04 LTS，提供稳定的Linux环境
• ROCM组件：包含rocm-dev开发包，满足基础ROCM开发需求
• 无内核依赖：不包含内核级组件，避免与宿主系统内核冲突
• 预配置仓库：已添加ROCM官方仓库及GPG密钥，便于后续包管理
• 精简安装：使用--no-install-recommends选项，减少冗余依赖
• 开发工具：包含build-essential等基础开发工具，支持编译需求

使用场景

• 作为基础镜像构建ROCM应用镜像
• 开发基于ROCM的深度学习框架或科学计算工具
• 构建自定义ROCM开发环境镜像

使用方法

1. 在Dockerfile中引用

下游镜像可通过FROM语句直接引用本镜像：

dockerfile
# 派生自本基础镜像
FROM [镜像名称]:[标签]

# 后续自定义配置（示例）
RUN apt-get update && apt-get install -y --no-install-recommends \
    rocm-libs \
    && rm -rf /var/lib/apt/lists/*

2. 本地运行（测试用）

若需本地运行镜像进行环境验证，可使用以下命令：

bash
docker run -it --rm [镜像名称]:[标签] /bin/bash

3. 构建信息参考

镜像构建时执行的关键步骤（基于原始Dockerfile）：

dockerfile
# 基础镜像
FROM ubuntu:20.04

# 安装依赖与配置仓库
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends curl libnuma-dev gnupg \
  && curl -sL http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | apt-key add - \
  && printf "deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main" | tee /etc/apt/sources.list.d/rocm.list \
  && apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
  sudo \
  libelf1 \
  rocm-dev \
  build-essential && \
  apt-get clean && \
  rm -rf /var/lib/apt/lists/*

注意事项

• 本镜像不含内核级组件，运行时需确保宿主系统已正确安装ROCM内核驱动
• 若需使用GPU功能，需在运行容器时添加--device=/dev/kfd --device=/dev/dri等设备映射参数
• 镜像中rocm-dev版本取决于构建时repo.radeon.com仓库中的最新版本，建议通过标签管理版本