Prerequisites

  • Architecture: x86_64
  • Chip models: MThreads MTT S5000
  • Host driver: 5.2.0-server
  • Container toolkit (optional) : KUAE Cloud Native Toolkits (MT Container Toolkit) >= 2.1.0

Image contents

Base image

ubuntu:22.04

System packages

Explicitly installed; the version is the one baked into this image:

  • 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 components

  • MUSA Toolkits 5.2.0
  • MCCL 2.4.0
  • muDNN 3.4.0

Environment

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

Launch

With the container toolkit (optional):

  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
  

Without a toolkit — plain 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
  

Verify

Inside the container, confirm the accelerator is visible:

  mthreads-gmi
  

Last updated: 2026-08-14 21:51:46 · 521f2968843d

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