前置条件

  • 架构: x86_64
  • 芯片型号: Iluvatar BI-V150
  • 宿主机驱动: 4.5.0
  • 容器工具包 (可选) : ix-container-toolkit >= 1.1.0

镜像内容

基于

harbor.baai.ac.cn/flagos-base/flagos-base-iluvatar-corex4.5.0:2.2.0 open_in_new

Python

3.12

主要 Python 软件包

  • flag_gems==5.4.0
  • flagtree==0.7.0+iluvatar3.6
  • numpy==1.26.4
  • torch==2.10.0+corex.4.5.0.20260804
  • torchaudio==2.10.0+corex.4.5.0.20260804
  • torchvision==0.25.0+corex.4.5.0.20260804
  • triton==3.2.0+corex.4.5.0.20260804

切换编译器

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

启动

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

  docker run --rm -it \
  --runtime iluvatar \
  --env IX_VISIBLE_DEVICES=all \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-iluvatar-corex4.5.0:2.2.0 bash
  

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

  docker run --rm -it \
  --device /dev/iluvatar0 \
  --device /dev/itrctl \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-iluvatar-corex4.5.0:2.2.0 bash
  

验证

在容器内,确认加速器可见:

  ixsmi
  

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