前置条件

  • 架构: x86_64
  • 芯片型号: Hygon BW1000
  • 宿主机驱动: 6.3.30-V1.4.1a
  • 容器工具包 (可选) : dcu-container-toolkit >= 1.3.0

镜像内容

基于

harbor.baai.ac.cn/flagos-base/flagos-base-hygon-dtk26.04:2.1.2 open_in_new

Python

3.10

主要 Python 软件包

  • flag_gems==5.3.4
  • flagtree==0.6.1+hcu3.6
  • flash_attn==2.8.3+das.opt1.dtk2604.torch290
  • numpy==1.26.4
  • torch==2.9.0+das.opt1.dtk2604
  • triton==3.5.1+das.opt1.dtk2604.torch290

切换编译器

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

启动

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

  docker run --rm -it \
  -e DCU_VISIBLE_DEVICES=all \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-hygon-dtk26.04:2.1.2 bash

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

  docker run --rm -it \
  --device /dev/kfd \
  --device /dev/mkfd \
  --device /dev/dri \
  --group-add video \
  -v /opt/hyhal:/opt/hyhal \
  --security-opt seccomp=unconfined \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-hygon-dtk26.04:2.1.2 bash

验证

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

  hy-smi
Edit this page

Last updated 16 Aug 2026, 12:55 +0800. history