NotImplementedError: Could not run ‘torchvision::nms’ with arguments from the ‘CUDA’ backend.

这个报错通常出现在 ComfyUI 执行涉及人脸检测/重绘的节点时(本 Issue 为 FaceDetailer),即 torchvision 的 NMS 算子没有在 CUDA 后端注册,而请求却发到了 CUDA。优先排查 CUDA 与 PyTorch/torchvision 的版本匹配情况。

快速结论:这个报错通常出现在 ComfyUI 执行涉及人脸检测/重绘的节点时(本 Issue 为 FaceDetailer),即 torchvision 的 NMS 算子没有在 CUDA 后端注册,而请求却发到了 CUDA。优先排查 CUDA 与 PyTorch/torchvision 的版本匹配情况。

适用环境:ComfyUI(Windows,从报错路径可确认运行在 Windows 系统、Python 3.12 安装目录);触发节点为 FaceDetailer;问题涉及 CUDA 后端与 torchvision 算子。Issue 未完整确认 Python/CUDA/PyTorch/torchvision 的具体版本组合与显卡型号。

最快修复方案:暂无确认的一步修复方案。Issue 中有一位用户反馈:在同类似问题下,将 CUDA 从 13.0 升级到 13.2,并安装最新 PyTorch 稳定版后恢复正常;但该反馈未由原报告者确认,可优先尝试。

注意事项:上述 CUDA + PyTorch 组合属于评论者的个人环境修复经验,并非官方结论,也未在原 Issue 中被复现验证;升级 CUDA 或 PyTorch 前请先记录当前版本,避免破坏已有工作环境。

问题场景

用户在 ComfyUI 中执行工作流时,Node ID 72 的 FaceDetailer 节点在运行过程中抛出异常,导致节点执行失败。

报错原文

NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). 'torchvision::nms' is only available for these backends: [CPU, Meta, BackendSelect, Python, ...]

原因分析

报错信息表明,torchvision::nms 算子只注册在 CPU 等后端(报错列表中 CPU 注册于 nms_kernel.cpp),而本次调用要求从 CUDA 后端执行,因此 PyTorch 找不到对应实现并抛出 NotImplementedError。可能原因是当前使用的 PyTorch/torchvision 构建与 CUDA 版本不匹配,导致 torchvision 的 CUDA 版 NMS 未被注册;也可能与自定义构建或选择性构建时遗漏该算子有关(报错原文本身提及此可能)。

环境排查

  • 确认当前 CUDA 版本,以及 PyTorch、torchvision 是否与 CUDA 版本匹配。
  • 确认 PyTorch 是否为官方稳定发行版,而非自定义/选择性构建。
  • 确认 Python 版本(报错路径显示为 Python 3.12)与上述组件的兼容性。
  • 在 ComfyUI 执行环境中验证 torchvision 的 CUDA 版本 NMS 是否可用。
  • 确认 FaceDetailer 节点及其相关依赖可正常加载。

解决步骤

  1. 先记录当前 Python、CUDA、PyTorch、torchvision 的版本,便于回滚或对比。
  2. 可优先尝试的修复:参照评论者经验,将 CUDA 升级至与目标 PyTorch 发行版匹配的版本(其曾从 13.0 升级到 13.2)。
  3. 安装与升级后 CUDA 对应的最新 PyTorch 稳定版本,并同步安装匹配的 torchvision。
  4. 重启 ComfyUI,替换使用同样环境的新进程重新加载模型与工作流。
  5. 重新运行 FaceDetailer 节点,观察是否仍出现 torchvision::nms 的 CUDA 后端报错。

验证方法

确认环境版本已更新后,在 ComfyUI 中重新执行包含 FaceDetailer 的工作流,确认节点正常完成、不再抛出 NotImplementedError;如仍报错,说明版本组合未解决问题,应回到环境排查继续核对 PyTorch 与 torchvision 的 CUDA 匹配情况。

参考来源

Comfy-Org/ComfyUI #17652

GamsGo AI

AI 工具推荐

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

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

了解 GamsGo AI

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

这个方案解决了吗?

celebrityanime
celebrityanime
文章: 34209

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注