DWPose: Onnxruntime not found 报错修复

以下教程参考 C站 大佬

在安装 comfyui_controlnet_aux 后可能会出现如下报错

UserWarning: DWPose: Onnxruntime not found or doesn't come with acceleration providers, switch to OpenCV with CPU device. DWPose might run very slowly
  warnings.warn("DWPose: Onnxruntime not found or doesn't come with acceleration providers, switch to OpenCV with CPU device. DWPose might run very slowly")

官方介绍中描述 NVidia 英伟达显卡所使用的版本默认支持 CUDA 11.8。如使用高于 CUDA 11.8 的版本可以自行编译 onnxruntime。

使用下面代码查看安装的是什么版本的 CUDA

cd $HOME/ComfyUI
source venv/bin/activate
python -c "import torch; print(torch.__version__); print(torch.version.cuda)"

如使用的是 NVidia CUDA 12.x 使用下面代码安装 onnxruntime

cd $HOME/ComfyUI
source venv/bin/activate
pip install onnxruntime-gpu

重启 ComfyUi 后发现 onnxruntime 已经安装上了

如何使用上面方法依然出现错误提示，可以尝试卸载 onnxruntime 后再重新安装，使用下面代码卸载

cd $HOME/ComfyUI
source venv/bin/activate
pip uninstall onnxruntime-gpu