【ComfyUI】Could not run ‘torchvision::nms’ with arguments from the ‘CUDA’ backend.

ComfyUIのカスタムノード「Impact Pack」にて

face_yolov8m.ptによる顔修正を試してみたところ、「Simple Detector for AnimateDiff (SEGS)」のノードで止まってしまいました。

エラー内容を見ると、

Error occurred when executing ImpactSimpleDetectorSEGS_for_AD:

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). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. ‘torchvision::nms’ is only available for these backends: [CPU, Meta, QuantizedCPU, BackendSelect, Python, FuncTorchDynamicLayerBackMode, Functionalize, Named, Conjugate, Negative, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradMPS, AutogradXPU, AutogradHPU, AutogradLazy, AutogradMeta, Tracer, AutocastCPU, AutocastCUDA, FuncTorchBatched, BatchedNestedTensor, FuncTorchVmapMode, Batched, VmapMode, FuncTorchGradWrapper, PythonTLSSnapshot, FuncTorchDynamicLayerFrontMode, PreDispatch, PythonDispatcher].

よく分からないので、アップデートやら色々やってみたけどさっぱり分からず悩んでググってたところ、下記サイトを発見。

pip listで見てみると・・・

同じ症状みたいなので、バージョンだけ0.17.2に合わせてインストール。

pip install torchvision==0.17.2+cu118 -f https://download.pytorch.org/whl/torch_stable.html

正常になった模様。

無事に次のノードへと流れました♪

参考サイト

とても助かりました。。。ありがとうございます<(_ _)>

YOLOv8でのエラー(Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.)を解決 - Qiita

状況公式サイトの通りPyTorch, Torchvision, TrochaudioをインストールYOLOv8モデルを実行するとfrom ultralytics import YOLOdat…

qiita.com