前置条件

  • 架构: x86_64
  • 芯片型号: MThreads MTT S5000
  • 宿主机驱动: 5.2.0-server
  • 容器工具包 (可选) : KUAE Cloud Native Toolkits (MT Container Toolkit) >= 2.1.0

镜像内容

基础镜像

ubuntu:22.04

系统软件包

显式安装；此处版本即为该镜像中实际打包的版本：

  • build-essential — 12.9ubuntu3
  • ca-certificates — 20260601~22.04.1
  • cmake — 3.22.1
  • curl — 7.81.0
  • g++ — 11.2.0
  • gcc — 11.2.0
  • git — 2.34.1
  • libelf1 — 0.186
  • libgfortran5 — 12.3.0
  • libnuma-dev — 2.0.14
  • libopenmpi-dev — 4.1.2
  • libpython3-dev — 3.10.6
  • openmpi-bin — 4.1.2
  • unzip — 6.0
  • vim — 8.2.3995

SDK 组件

  • MUSA Toolkits 5.2.0
  • MCCL 2.4.0
  • muDNN 3.4.0

环境变量

  • MUSA_HOME=/usr/local/musa
  • PATH=/usr/local/musa/bin:${PATH}
  • LD_LIBRARY_PATH=/usr/local/musa/lib

启动

使用容器工具包 (可选)：

  docker run --rm -it \
  --runtime mthreads \
  --env MTHREADS_VISIBLE_DEVICES=all \
  harbor.baai.ac.cn/flagos-base/flagos-base-mthreads-musa5.2.0:2.1.2 bash

无需工具包 —— 直接使用 docker / podman：

  docker run --rm -it \
  --device /dev/mtgpu.0 \
  --device /dev/dri \
  -v /usr/bin/mthreads-gmi:/usr/bin/mthreads-gmi:ro \
  harbor.baai.ac.cn/flagos-base/flagos-base-mthreads-musa5.2.0:2.1.2 bash

验证

在容器内，确认加速器可见：

  mthreads-gmi

更新时间: 2026-08-14 21:51:46 · 521f2968843d

Edit this page

Last updated 15 Aug 2026, 09:57 +0800. history