megatron_training0.18.2-metax-maca3.8.1.3

Prerequisites link

• Architecture: x86_64
• Chip models: MetaX C550
• Host driver: 3.9.6
• Container toolkit (optional) ⓘ: metax-docker >= 0.15.3

Image contents link

Built on link

Python link

3.12

Application package link

megatron-core[training]==0.18.2

Launch link

Published: harbor.baai.ac.cn/flagos-app/megatron_training0.18.2-metax-maca3.8.1.3:2.2.0-0.3.0

The image name is long — assign it to a variable first:

  IMG=harbor.baai.ac.cn/flagos-app/megatron_training0.18.2-metax-maca3.8.1.3:2.2.0-0.3.0
  

The two approaches below are alternatives — pick the one that matches how your host runs containers:

With the container toolkit link

Start an interactive shell:

  metax-docker run --rm -it $IMG bash
  

Start the app with its default settings:

  metax-docker run --rm -it $IMG
  

Pass arguments to the launcher:

  metax-docker run --rm -it $IMG megatron-train --model-type GPT
  

Without a toolkit — plain docker / podman link

Start an interactive shell:

  docker run --rm -it \
  --device /dev/mxcd \
  --device /dev/dri \
  --group-add video \
  $IMG bash
  

Start the app with its default settings:

  docker run --rm -it \
  --device /dev/mxcd \
  --device /dev/dri \
  --group-add video \
  $IMG
  

Pass arguments to the launcher:

  docker run --rm -it \
  --device /dev/mxcd \
  --device /dev/dri \
  --group-add video \
  $IMG megatron-train --model-type GPT