Tencent / MimicMotion

High-Quality Human Motion Video Generation with Confidence-aware Pose Guidance
https://tencent.github.io/MimicMotion/
Other
1.46k stars 114 forks source link

环境依赖建议加上onnxruntime-gpu #44

Open kingydw opened 1 month ago

kingydw commented 1 month ago

如果dwpose阶段有UserWarning: Specified provider 'CUDAExecutionProvider' is not in available provider names.Available providers: 'AzureExecutionProvider, CPUExecutionProvider',说明onnx没有用上gpu,可以如下double check

import onnxruntime as ort
print("Available providers:", ort.get_available_providers())

解决: pip install onnxruntime-gpu

效果 dwpose提取阶段,从4s/iter 变成 30iter/s,提升120倍 dwpose使用cpu

dwpose使用cpu

dwpose使用cuda后

dwpose使用gpu
wwfcnu commented 1 month ago

我安装了pip install onnxruntime-gpu,但是还没有CUDAExecutionProvider,只有'AzureExecutionProvider, CPUExecutionProvider'

yuananf commented 1 month ago

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