前置条件

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

镜像内容

基于

harbor.baai.ac.cn/flagos-runtime/flagos-runtime-mthreads-musa4.3.6:2.2.0 open_in_new

Python

3.10

应用软件包

megatron-core[training]==0.17.1

启动

已发布: harbor.baai.ac.cn/flagos-app/megatron_training0.17.1-mthreads-musa4.3.6:2.2.0-0.2.3

镜像名较长——先将其设为变量：

  IMG=harbor.baai.ac.cn/flagos-app/megatron_training0.17.1-mthreads-musa4.3.6:2.2.0-0.2.3

以下两种方式任选其一：

使用容器工具包

启动交互式 shell：

  docker run --rm -it \
  --runtime mthreads \
  --env MTHREADS_VISIBLE_DEVICES=all \
  $IMG bash

以默认设置启动应用：

  docker run --rm -it \
  --runtime mthreads \
  --env MTHREADS_VISIBLE_DEVICES=all \
  $IMG

向启动器传参：

  docker run --rm -it \
  --runtime mthreads \
  --env MTHREADS_VISIBLE_DEVICES=all \
  $IMG megatron-train --model-type GPT

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

启动交互式 shell：

  docker run --rm -it \
  --device /dev/mtgpu.0 \
  --device /dev/dri \
  -v /usr/bin/mthreads-gmi:/usr/bin/mthreads-gmi:ro \
  $IMG bash

以默认设置启动应用：

  docker run --rm -it \
  --device /dev/mtgpu.0 \
  --device /dev/dri \
  -v /usr/bin/mthreads-gmi:/usr/bin/mthreads-gmi:ro \
  $IMG

向启动器传参：

  docker run --rm -it \
  --device /dev/mtgpu.0 \
  --device /dev/dri \
  -v /usr/bin/mthreads-gmi:/usr/bin/mthreads-gmi:ro \
  $IMG megatron-train --model-type GPT
Edit this page

Last updated 24 Sep 2026, 13:40 +0800. history