babysor / MockingBird

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

开始使用CPU训练 现在该如何改用GPU训练 尝试各种办法都不行 #670

Closed b95595 closed 2 years ago

b95595 commented 2 years ago

Using model: Tacotron Using device: cpu

VERT2022 commented 2 years ago

安装gpu版本的pytorch,电脑系统的cuda版本要和pytorch里面的一致 python里面执行torch.cuda.is_available()后,结果为True就可以了,建议安装pytorch 1.11.0 + cuda 11.5