将 facebook/deit-small-distilled-patch16-224 模型适配至华为昇腾 NPU(Ascend910B)。通过 ModelScope 的 snapshot_download 功能下载权重文件,并使用 transformers 库中的 DeiTForImageClassificationWithTeacher 类加载本地权重。
由于普通 site-packages 目录不可写,默认使用用户安装模式 已满足要求:torch 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 -r requirements.txt 第 1 行)(2.9.0+cpu) 已满足要求:torchvision 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 -r requirements.txt 第 2 行)(0.24.0) 已满足要求:transformers 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 -r requirements.txt 第 3 行)(4.57.6) 已满足要求:pillow 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 -r requirements.txt 第 4 行)(12.2.0) 已满足要求:requests 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 -r requirements.txt 第 5 行)(2.33.1) 已满足要求:numpy 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 -r requirements.txt 第 6 行)(1.26.4) 已满足要求:huggingface_hub 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 -r requirements.txt 第 7 行)(0.36.2) 已满足要求:accelerate 位于 /opt/atomgit/.local/lib/python3.11/site-packages(来自 -r requirements.txt 第 8 行)(1.13.0) 已满足要求:modelscope 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 -r requirements.txt 第 9 行)(1.35.3) 已满足要求:timm 位于 /opt/atomgit/.local/lib/python3.11/site-packages(来自 -r requirements.txt 第 10 行)(1.0.27) 已满足要求:safetensors 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 -r requirements.txt 第 11 行)(0.7.0) 已满足要求:filelock 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 torch->-r requirements.txt 第 1 行)(3.25.2) 已满足要求:typing-extensions>=4.10.0 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 torch->-r requirements.txt 第 1 行)(4.15.0) 已满足要求:sympy>=1.13.3 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 torch->-r requirements.txt 第 1 行)(1.14.0) 已满足要求:networkx>=2.5.1 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 torch->-r requirements.txt 第 1 行)(3.6.1) 已满足要求:jinja2 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 torch->-r requirements.txt 第 1 行)(3.1.6) 已满足要求:fsspec>=0.8.5 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 torch->-r requirements.txt 第 1 行)(2026.3.0) 已满足要求:packaging>=20.0 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 transformers->-r requirements.txt 第 3 行)(26.0) 已满足要求:pyyaml>=5.1 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 transformers->-r requirements.txt 第 3 行)(6.0.3) 已满足要求:regex!=2019.12.17 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 transformers->-r requirements.txt 第 3 行)(2026.3.32) 已满足要求:tokenizers<=0.23.0,>=0.22.0 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 transformers->-r requirements.txt 第 3 行)(0.22.2) 已满足要求:tqdm>=4.27 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 transformers->-r requirements.txt 第 3 行)(4.67.3) 已满足要求:hf-xet<2.0.0,>=1.1.3 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 huggingface_hub->-r requirements.txt 第 7 行)(1.4.3) 已满足要求:charset_normalizer<4,>=2 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 requests->-r requirements.txt 第 5 行)(3.4.6) 已满足要求:idna<4,>=2.5 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 requests->-r requirements.txt 第 5 行)(3.11) 已满足要求:urllib3<3,>=1.26 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 requests->-r requirements.txt 第 5 行)(2.6.3) 已满足要求:certifi>=2023.5.7 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 requests->-r requirements.txt 第 5 行)(2026.2.25) 已满足要求:psutil 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 accelerate->-r requirements.txt 第 8 行)(7.2.2) 已满足要求:setuptools 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 modelscope->-r requirements.txt 第 9 行)(79.0.1) 已满足要求:mpmath<1.4,>=1.1.0 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 sympy>=1.13.3->torch->-r requirements.txt 第 1 行)(1.3.0) 已满足要求:MarkupSafe>=2.0 位于 /usr/local/python3.11.14/lib/python3.11/site-packages(来自 jinja2->torch->-r requirements.txt 第 1 行)(3.0.3) [LOG_WARNING] 无法创建目录,目录:/home/atomgit/ascend/log,可能原因:没有该文件或目录。路径字符串为空路径字符串为空正在从 https://www.modelscope.cn 下载模型至目录:/opt/atomgit/.cache/modelscope/hub/models/facebook/deit-small-distilled-patch16-224 Top-1:valley, vale(0.7226) Top-2:lakeside, lakeshore(0.2106) Top-3:alp(0.0193) Top-4:promontory, headland, head, foreland(0.0088) Top-5:volcano(0.0086)
推理结果(NPU Top-5):
日志保存在 。
对单张测试图片进行 CPU 与 NPU 一致性验证:
| 指标 | 数值 |
|---|---|
| max_abs_error | 0.023058 |
| mean_abs_error | 0.003944 |
| relative_error | 0.4435% |
| cosine_similarity | 0.999994 |
| threshold | 1.0% |
| 结果 | PASS |
| 指标 | 数值 |
|---|---|
| Avg latency | 6.20 ms |
| Throughput | 161.37 images/sec |
本项目包含单图 smoke consistency 验证,非官方 ImageNet 完整验证集评测。详细指标见第 4 节。
截图保存在 。