前置条件

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

镜像内容

基于

harbor.baai.ac.cn/flagos-base/flagos-base-ascend-cann9.0.0-910c:2.2.0 open_in_new

Python

3.11

主要 Python 软件包

  • cann-shmem==1.6.0
  • flag_gems==5.4.0
  • flagtree==0.7.0+ascend3.5
  • numpy==1.26.4
  • torch-npu==2.10.0
  • torch==2.10.0+cpu
  • torchaudio==2.10.0+cpu
  • torchvision==0.25.0+cpu
  • triton==3.5.0 (+ triton_ascend==3.2.1)

切换编译器

本镜像同时包含 FlagTree（默认）和 Triton。在容器内执行 compiler triton 可切换到 Triton，执行 compiler flagtree 切回，执行 compiler 查看当前编译器。

启动

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

  docker run --rm -it \
  --privileged \
  -e ASCEND_VISIBLE_DEVICES=0,1 \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-ascend-cann9.0.0-910c:2.2.0 bash

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

  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 \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-ascend-cann9.0.0-910c:2.2.0 bash

验证

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

  npu-smi info
Edit this page

Last updated 26 Sep 2026, 01:52 UTC. history