变更摘要

本 PR 主要将评测/开发镜像的默认 CANN 版本从 9.0.1（docker/dev 为已下架的 9.0.0）统一提升到 9.1.0，并把 torch_npu 同步从 2.10.0.post2 升级为与 CANN 9.1.0 配套的 2.10.0.post4。同时修正了报告中 docker 字段长期硬编码为 "cake-ci / CANN 9.0.0" 的问题，改为由镜像注入的 CANN_BENCH_IMAGE、容器探测与真实 CANN 探测值共同决定，使报告字段与实际运行环境一致，并补充了对应单元测试。

主要改动

• 默认 CANN 版本升至 9.1.0：docker/base/Dockerfile 的 ARG CANN_VERSION、docker/eval/Dockerfile 的 BASE_IMAGE、docker/eval/build.sh 的 CANN_VERSION 及 docker/dev/Dockerfile 的 ARG CANN_VERSION 均由旧版本改为 9.1.0，相应 run.sh 中的默认镜像 tag 也同步更新。
• torch_npu 升级到 2.10.0.post4：docker/base/pyproject.toml、docker/eval/pyproject.toml 和根 pyproject.toml 将 torch-npu==2.10.0.post2（根为 2.10.0）改为 2.10.0.post4，以匹配官方配套表中 torch_npu 2.10 仅对应 CANN 9.1.0 的关系。
• docker 报告字段改为真实取值：setup_info.py 中的 _detect_docker() 不再硬编码，新增 _in_container() 后按 CANN_BENCH_IMAGE 环境变量 → 容器内回退 "container" → 裸机返回 None 三级取值，CANN 版本统一取自 _get_cann_version()。
• 构建流程注入镜像标识：docker/eval/build.sh 新增 --build-arg "CANN_BENCH_IMAGE=${IMAGE}"，docker/eval/Dockerfile 新增 ARG CANN_BENCH_IMAGE 与对应 ENV，使报告能显示实际产出镜像的 tag；report_template.html 中的示例值也同步更新。
• docker/dev 镜像参数与文档修正：新增 ARG VARIANT=devel 并把基础镜像改为 cann:${CANN_VERSION}-${DEVICE}-ubuntu22.04-py3.12-${VARIANT}，同时更正了可用 DEVICE、CANN_VERSION 取值及镜像 tag 格式的注释；新增 tests/ut/test_setup_info.py 覆盖 docker 字段的三条取值路径。

代码审查

✅ 未发现问题

Thanks for your pull-request.
The full list of commands accepted by me can be found at here.
You can get sig-info at here.
You can self-configure the PR merge rules for this repository. For more details, please refer to Here.
For more, you also can visit HICANN.

PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

💡 Tip:

• Committer can comment /approve or /lgtm
• Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

vINyLogY, thanks for your pull request. All authors of the commits have signed the CLA. 👍

compile

compile

流水线任务触发成功
任务链接 [716e5b3584814328a06c92731e4d76f1][流水线指导]

[2026-08-18 15:29:52]    CI执行结束

这里还有一个版本配套问题需要修正：

docker/dev/Dockerfile:20-24 说明 CANN 9.1.0 对应 torch_npu 2.10.0.post4，并声明基础镜像已自带 torch/torch_npu、此处不再钉版本；但 docker/dev/Dockerfile:69-70 实际仍会执行：

RUN uv pip install --system --no-deps --index-url https://download.pytorch.org/whl/cpu torch==2.10.0
RUN uv pip install --system --no-deps --index-url ${PYPI_INDEX_URL:-https://mirrors.huaweicloud.com/repository/pypi/simple/} torch_npu==2.10.0

torch_npu==2.10.0 是精确版本约束，不会匹配 2.10.0.post4，而且会覆盖基础镜像中的配套版本。因此默认 docker/dev 构建产物与本 PR 的兼容性说明不一致，并可能重新引入 CANN/torch_npu ABI 配套风险。

建议二选一：

1. 按注释所述，删除这里对 torch/torch_npu 的覆盖安装，直接使用基础镜像自带版本，并在 test_env.py 中校验实际版本；或
2. 如果确实需要显式安装，则改为官方配套的精确版本（例如本 PR 引用的 torch_npu==2.10.0.post4），并同步说明与版本 smoke test。

修复后建议实际构建默认 docker/dev 镜像，并回读 torch.__version__、torch_npu.__version__ 与 CANN 版本，避免只验证 Dockerfile 文本。

Notification

This pull request has been changed(code update) or closed, so removes the following label(s): ci-pipeline-passed.

三条评审意见都已修,commit 0d329e27。

@Deng_Pan — docker/dev 的 torch/torch_npu 配套

判断对:头注释("torch/torch_npu 由基础镜像自带,不在此处钉")与 docker/dev/Dockerfile:69-70 的 torch==2.10.0 / torch_npu==2.10.0 自相矛盾。错的是注释。

按你的建议先去核实底座到底带不带 —— 拉下默认基础镜像 ascendhub/cann:9.1.0-910b-ubuntu22.04-py3.12-devel(linux/arm64)实跑:

Python 3.12.13
python3 -c 'import torch'      -> ModuleNotFoundError: No module named 'torch'
python3 -c 'import torch_npu'  -> ModuleNotFoundError: No module named 'torch_npu'
pip list | grep -c '^torch'    -> 0

底座一个 torch 分发都没有(另一个已在本地的 cann:9.1.0-beta.3-950-ubuntu22.04-py3.11 同样为 0)。所以建议 1(删掉覆盖安装、用底座自带)会直接产出一个没有 torch 的镜像,走的是建议 2:

• TORCH_VERSION=2.10.0 / TORCH_NPU_VERSION=2.10.0.post4 提成 ARG,两条 uv pip install 改用它们;
• 两个值落到 ENV CANN_BENCH_TORCH_VERSION / CANN_BENCH_TORCH_NPU_VERSION;
• test_env.py 的 [1] 回读实际装到的分发版本与 ENV 比对,不一致即 FAIL。用 importlib.metadata.version() 而不是 __version__,因为 pin 作用在分发版本上;+cpu 这类 local label 不属于配套关系,比对前剥掉。

这样"文本"和"产物"就被一条硬断言绑住了:钉错、或被上游覆盖成别的串号,run.sh smoke 直接红。

按你说的"避免只验证 Dockerfile 文本",实际建了默认镜像(原生 linux/arm64,BUILD_EXIT=0)并回读:

ENV pins:  CANN_BENCH_TORCH_VERSION=2.10.0  CANN_BENCH_TORCH_NPU_VERSION=2.10.0.post4
installed: torch 2.10.0+cpu | torch-npu 2.10.0.post4
CANN:      Version=9.1.0

即落在官方配套表 9.1.0 的那一行。

一处如实说明:构建这台是 mac,没有 NPU 驱动,import torch_npu 在这里必然失败(ImportError: libascend_hal.so,/usr/local/Ascend/driver 目录不存在),因此 test_env.py 的 [1]-[3] 没法在此跑完 —— 上面那组数是绕开 import、直接读分发版本与 ENV 得到的。完整 smoke 仍需在 NPU 机器上补一次。

@su-yueming [P1] — 回退配套说明

删掉了。docker/base/README.md 里"回退到 9.0.1 时把 torch_npu 改回 2.10.0.post2"整段移除,换成中性的一句:换版本时按官方配套表重新选串号、重新 uv lock,表里没有对应行的 CANN 版本不要用。同一节上面的 换 CANN 版本 示例原本正好拿 9.0.1 举例,一并换成 <版本> 占位符,免得删了正文又留一个变相推荐。

@su-yueming [P3] — 过时的"硬编码"说明

删掉了 docs/design/report_design.md 的 (硬编码,后续改软编码)。顺带:上一行改成 docker ← $CANN_BENCH_IMAGE + 实测 CANN 版本 之后已经撑破 ASCII 框,拆成两行对齐;同一个框里 framework / license 两行本来就各差一格(改动前就有),一并补齐。

另一处同源的陈旧文本

docker/eval/README.md 「已知取舍」里"底座的 torch_npu==2.10.0.post2 与仓库 pyproject.toml 钉的 2.10.0 不是同一串号,沿用底座的 post2"—— 本 PR 之后三处 pyproject.toml 都是 2.10.0.post4,这条取舍已不存在,删除。

compile

流水线任务触发成功
任务链接 [090d798efc7d43db9ca9359feb4683ef][流水线指导]

[2026-08-24 10:08:23]    CI执行结束

@Deng_Pan 补上 NPU 实机复验。上一条那组数是在 mac 上跑的(没有驱动,import torch_npu 挂在 libascend_hal.so,[1]-[3] 跑不完)。现在在 q7(910B2 / aarch64 / driver 25.5.1 / npu-smi 25.5.1)上,按默认参数重建了 docker/dev 并跑完整 smoke:

$ docker build -t cann-bench:cann9.1.0-910b-latest .     # 不带任何 build-arg
Successfully tagged cann-bench:cann9.1.0-910b-latest

$ bash run.sh smoke
[OK]   [1] python 3.12.13, torch 2.10.0+cpu, torch_npu 2.10.0.post4
[OK]   [2] torch_npu.npu.device_count() = 8
[OK]   [3] npu-smi reachable
[OK]   [4] CANN compiler Version=9.1.0
[SKIP] [5] Triton-Ascend not requested for this image

ALL CHECKS PASSED

即 CANN 9.1.0 <-> torch 2.10.0 <-> torch_npu 2.10.0.post4,正是官方配套表那一行;[1] 的新断言在实机上通过。

新断言不是摆设 —— 同一个镜像,只把 ENV 改成表外的 post2,[1] 立刻红、退出码 1:

$ docker run ... -e CANN_BENCH_TORCH_NPU_VERSION=2.10.0.post2 ... python3 /test_env.py
[FAIL] [1] import/version: torch-npu 2.10.0.post4, image pins 2.10.0.post2
[OK]   [2] torch_npu.npu.device_count() = 8
[OK]   [3] npu-smi reachable
[OK]   [4] CANN compiler Version=9.1.0
FAILED: [1]

也就是说:以后谁把 Dockerfile 里的串号改错、或上游把版本换掉,run.sh smoke 会直接拦下来,不再是靠注释约束。

另外"AscendHub 底座不带 torch"这条在 q7 上又多了两个样本,连同之前两个,四个 ascendhub/cann:* 镜像的 pip list | grep -c '^torch' 全是 0:

所以走"显式安装 + 钉配套串号"这条是必须的,建议里的方案 1 会产出一个没有 torch 的镜像。

complie

/approve
/lgtm

/lgtm