前置条件

  • 架构: x86_64
  • 芯片型号: Kunlunxin P800
  • 宿主机驱动: 5.37.1
  • 容器工具包 (可选) : xpu_container >= 1.0.13

镜像内容

基于

harbor.baai.ac.cn/flagos-base/flagos-base-kunlunxin-xre5.37.1:2.1.2 open_in_new

Python

3.10

主要 Python 软件包

  • apex==0.1
  • benchflow==1.0.0
  • colorama==0.4.6
  • flag_gems==5.3.4
  • flagtree==0.6.1+xpu3.6
  • flash_attn==2.4.2+7e2dd4d
  • hydrax==0.1.0
  • hyperparameter==0.5.6
  • numpy==2.2.6
  • psutil==6.1.0
  • regex==2026.4.4
  • torch==2.9.0+cu129
  • torch_plugin==0.1.0
  • torch_xray==2.0.4
  • torchaudio==2.9.0+cu129
  • torchvision==0.24.0+cu129
  • triton==3.0.0+a48aedef
  • xformers==0.0.29+1e7a8ec.d20260114
  • xmlir==1.0.0.1

切换编译器

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

启动

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

  docker run --rm -it \
  --runtime xpu \
  -e CXPU_VISIBLE_DEVICES=0 \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-kunlunxin-xre5.37.1:2.1.2 bash

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

  docker run --rm -it \
  --device /dev/xpu0 \
  --device /dev/xpuctrl \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-kunlunxin-xre5.37.1:2.1.2 bash

验证

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

  xpu-smi
Edit this page

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