前置条件

  • 架构: x86_64
  • 芯片型号: Enflame Zixiao C200 (S60)
  • 宿主机驱动: 1.10.6
  • 容器工具包 (可选) : tencent-container-toolkit >= 2.0.52

镜像内容

基于

harbor.baai.ac.cn/flagos-base/flagos-base-enflame-tops1.10.6:2.1.2 open_in_new

Python

3.12

主要 Python 软件包

  • enflame-modelopt==3.6.20260615+torch.2.11.0
  • flag_gems==5.3.4
  • flagtree==0.6.1+enflame3.6
  • flash-attn==2.7.2+torch.2.11.0.gcu.3.8.20260706
  • numpy==2.3.5
  • pyefml==1.10.6
  • tops-extension==3.6.20260529+torch.2.11.0
  • topstx==1.10.6
  • torch-gcu==2.11.0+3.8.20260713
  • torch==2.11.0+cpu
  • torchaudio==2.11.0+cpu
  • torchcodec==3.8.20260624+torch.2.11.0
  • torchvision==0.26.0+cpu
  • triton==3.6.0 (+ triton-gcu==3.6.0+1.0.20260722)

切换编译器

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

环境变量

  • TORCH_GCU_ENABLE_INT64_AND_UINT64=1
  • ENABLE_I64_CHECK=0
  • TORCHGCU_INDUCTOR_ENABLE=1
  • ENFLAME_PT_OP_DEBUG_CONFIG=fallback_cpu=all_ops

启动

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

  docker run --rm -it \
  --network host \
  -e ENFLAME_VISIBLE_DEVICES=all \
  -e TENCENT_VISIBLE_DEVICES=all \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-enflame-tops1.10.6:2.1.2 bash

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

  docker run --rm -it \
  --privileged \
  -v /dev:/dev \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-enflame-tops1.10.6:2.1.2 bash

验证

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

  efsmi
Edit this page

Last updated 15 Aug 2026, 12:51 +0800. history