sunrise-tangrt1.2.0

前置条件 link

• 架构: x86_64
• 芯片型号: Sunrise SR-SUN-S2-X1
• 宿主机驱动: 0.24.0

镜像内容 link

基于 link

Python link

3.10

主要 Python 软件包 link

• flag_gems==5.3.4
• flagtree==0.6.0+sunrise3.6
• numpy==2.2.6
• torch-ptpu==0.2.3+torch2.11
• torch==2.11.0+cpu
• torchaudio==2.11.0+cpu
• torchvision==0.26.0+cpu
• triton==3.6.0.1+git0a5cfb35

切换编译器 link

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

启动 link

启动交互式 shell（docker 或 podman 均可）：

  docker run --rm -it \
  --privileged \
  -v /dev:/dev \
  harbor.baai.ac.cn/flagos-runtime/flagos-runtime-sunrise-tangrt1.2.0:2.1.2 bash
  

验证 link

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

  pt_smi