Open kingydw opened 1 month ago
我安装了pip install onnxruntime-gpu,但是还没有CUDAExecutionProvider,只有'AzureExecutionProvider, CPUExecutionProvider'
cuda12的onnxruntime-gpu安装不太一样
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
参考: https://onnxruntime.ai/docs/install/#install-onnx-runtime-gpu-cuda-12x
如果dwpose阶段有
UserWarning: Specified provider 'CUDAExecutionProvider' is not in available provider names.Available providers: 'AzureExecutionProvider, CPUExecutionProvider'
,说明onnx没有用上gpu,可以如下double check解决:
pip install onnxruntime-gpu
效果 dwpose提取阶段,从4s/iter 变成 30iter/s,提升120倍 dwpose使用cpu
dwpose使用cuda后