我想在本地训练模型，运行后出现CPU training is not allowed.

SayaSS / vits-finetuning

Fine-Tuning your VITS model using a pre-trained model

MIT License

546 stars 86 forks source link

Closed Apoil closed 1 year ago

Apoil commented 1 year ago

这是设置的问题，还是本地显卡基础配置的问题？我该如何使用GPU训练，

SayaSS commented 1 year ago

需要NVIDIA显卡以及cuda环境，并且torch能支持GPU

Apoil commented 1 year ago

需要NVIDIA显卡以及cuda环境，并且torch能支持GPU

好的谢谢