search

babysor / MockingBird

🚀AI拟声: 5秒内克隆您的声音并生成任意语音内容 Clone a voice in 5 seconds to generate arbitrary speech in real-time
Other
34.57k stars 5.15k forks source link

为什么main的新版本，我开启训练后只能使用CPU，没法使用GPU，训练速度低到 0.08setp/s #902

Open syy7854 opened 1 year ago

syy7854 commented 1 year ago

为什么main的新版本，我开启训练后只能使用CPU，没法使用GPU，训练速度低到 0.08setp/s

{| Epoch: 52/3334 (3/3) | Loss: 1.005 | 0.081 steps/s | Step: 0k | } {| Epoch: 53/3334 (3/3) | Loss: 0.9955 | 0.081 steps/s | Step: 0k | } {| Epoch: 54/3334 (3/3) | Loss: 0.9905 | 0.081 steps/s | Step: 0k | } {| Epoch: 55/3334 (3/3) | Loss: 0.9797 | 0.081 steps/s | Step: 0k | }

任务管理器中，GPU显示为3%，CPU为60%

是还需要修改什么参数吗

babysor commented 1 year ago

pytorch gpu版没安装对，查看下

HaSaKiYasuooo commented 1 year ago

怎样才算安装正确，如何匹配！

L1590 commented 12 months ago

import torch torch.cuda.is_available() 若返回为True，则使用的是GPU版本的torch，若为False，则为CPU版本