search

anliyuan / Ultralight-Digital-Human

一个超轻量级、可以在移动端实时运行的数字人模型
933 stars 148 forks source link

推理时能用GPU吗 #74

Open Wiphone opened 15 hours ago

Wiphone commented 15 hours ago

推理时，使用文档中的语句：

python inference.py --asr hubert --dataset ./your_data_dir/ --audio_feat your_test_audio_hu.npy --save_path xxx.mp4 --checkpoint your_trained_ckpt.pth

image

但是全部用的是CPU去处理，GPU没动静，是需要配置参数吗

ghfyxw commented 14 hours ago

我也遇到了同样的问题，CPU使用率到100%，不知道楼主找到问题所在了吗

emrahtoy commented 5 hours ago

@ghfyxw @Wiphone for ML training and such jobs, you should change the graph ( on the top right side ) from "Copy" to "Cuda". Also, GPU Memory usage could be seen as an indicator.