No module named ‘torch_npu’

问题场景

用户在 ComfyUI 中运行一个仅包含原生节点的 Generate Text 工作流，向节点传入包含多个字符串的列表作为 prompt。第一个 prompt 能正常处理，处理第二个 prompt 时 ComfyUI 崩溃，并由 systemd 生成 coredump。同时，启动日志中出现 Ascend 后端探测失败的提示。

报错原文

[INFO] Found comfy_kitchen backend ascend: {'available': False, 'disabled': False, 'unavailable_reason': "torch-npu is not installed: No module named 'torch_npu'", 'capabilities': []}

原因分析

关于 No module named 'torch_npu'：这是 ComfyUI 启动时枚举可用计算后端的结果。Ascend 后端需要 torch-npu 包，当前环境未安装，因此该后端被标记为不可用。对于使用 CUDA 的环境，这属于正常的不可用提示。

关于实际的崩溃：Issue 讨论中的结论指向多字符串列表 prompt 的批处理过程：第二次运行时试图复用第一次运行中已释放的图（freed graph），因此触发崩溃。该结论来自讨论区的判断；Issue 中未给出代码级修复或发布版本。

环境排查

• 确认是否真的使用 Ascend/NPU 硬件；若为 NVIDIA 显卡，可忽略该 torch_npu 提示。
• 确认 PyTorch 版本与 CUDA 版本匹配（Issue 中为 torch 2.14.1+cu130）。
• 确认 Python 版本（Issue 中为 3.13）。
• 确认已禁用自定义节点并复现问题，以排除第三方节点干扰。
• 确认 ComfyUI 版本；如果问题出在图生命周期管理上，则需要更新到包含相关修复的版本。

解决步骤

1. 先区分两类信息：No module named 'torch_npu' 是 Ascend 后端探测的提示；如果后续日志显示 cuda 后端 available，则它与本次崩溃无直接关系，无需处理。
2. 使用 Issue 中提供的原生工作流（generate_text_bug.json）复现：向 Generate Text 节点传入包含多个字符串的列表 prompt。
3. 如果只是要临时绕过崩溃，可优先尝试拆分列表：将多字符串列表拆成多条单独运行，避免在同一图中连续处理多个 prompt（此做法为推测性规避，Issue 未验证）。
4. 检查并更新 ComfyUI 到最新版本，确认是否已修复“第二次运行复用已释放图”的问题。
5. 如果仍复现，可在 Issue 中补充完整 traceback 和 coredump 信息，便于定位图释放时机。

验证方法

重新运行同一工作流：观察处理第二个 prompt 时是否还出现 coredump；若不再崩溃且能输出与输入列表长度对应的字符串列表，即视为解决。同时确认启动日志中的 CUDA 后端仍为 available，且无需依赖 torch_npu。

参考来源

Comfy-Org/ComfyUI #17167

AI 工具推荐

想把多个 AI 模型放在一个入口？

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型，适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接：通过此链接购买，我可能获得佣金，不影响你的价格。