前置条件

  • 架构: aarch64
  • 芯片型号: Ascend 910B
  • 宿主机驱动: 26.0.rc1
  • 容器工具包 (可选) : Ascend-docker-runtime >= 6.0.RC3

镜像内容

基于

harbor.baai.ac.cn/flagos-runtime/flagos-runtime-ascend-cann9.0.0:2.2.0 open_in_new

Python

3.11

应用软件包

megatron-core[rl]==0.17.1

启动

已发布: harbor.baai.ac.cn/flagos-app/megatron_rl0.17.1-ascend-cann9.0.0:2.2.0-0.2.3

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

  IMG=harbor.baai.ac.cn/flagos-app/megatron_rl0.17.1-ascend-cann9.0.0:2.2.0-0.2.3

以下两种方式任选其一：

使用容器工具包

启动交互式 shell：

  docker run --rm -it \
  --privileged \
  -e ASCEND_VISIBLE_DEVICES=0,1 \
  $IMG bash

暂无启动器。 该镜像尚未提供启动器或默认命令——可先启动交互式 shell 查看镜像内容；启动器将随应用的入口点一并提供。

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

启动交互式 shell：

  docker run --rm -it \
  --device /dev/davinci0 \
  --device /dev/davinci1 \
  --device /dev/davinci_manager \
  --device /dev/devmm_svm \
  --device /dev/hisi_hdc \
  -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
  -v /usr/local/dcmi:/usr/local/dcmi \
  -v /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi \
  $IMG bash

暂无启动器。 该镜像尚未提供启动器或默认命令——可先启动交互式 shell 查看镜像内容；启动器将随应用的入口点一并提供。

Edit this page

Last updated 25 Sep 2026, 10:04 +0800. history