昇腾910b,mineru 2.5.0 + pipeline 搭配哪个版本的torch_npu?
Closed this issue · 1 comments
penond commented
请问,昇腾910b,mineru 2.5.0 + pipeline 搭配哪个版本的torch_npu?
dosubot commented
昇腾910B + MinerU 2.5.0 + pipeline 推荐使用 torch 2.6.0 和 torch_npu 2.6.0rc1(需匹配你的 Python 版本和架构)。torch/torchvision 可以直接用 PyPI 官方包,torch_npu 必须从昇腾官方 Gitee 仓库下载对应 whl 包,否则无法用 NPU 加速。升级 torch/torch_npu 后可避免安全漏洞(CVE-2025-32434),并保证 NPU 能被正常识别和利用。社区已验证该组合可用,安装命令示例:
pip install torch==2.6.0 torchvision==0.21.0
pip install torch_npu-2.6.0rc1-cp310-cp310-manylinux_2_28_aarch64.whl如需详细步骤或 Dockerfile,可参考相关 issue 说明和社区经验。
注意:pipeline 模式下 NPU 加速依赖 torch_npu、驱动、固件等环境配置完整,否则容易回退到 CPU。多进程并发时,务必每进程独立初始化模型,避免 fork 前加载,详见并发最佳实践。
To reply, just mention @dosu.
How did I do? Good | Irrelevant | Incorrect | Verbose | Hallucination | Report 🐛 | Other