[babysor/MockingBird]关于能不能使用gpu训练

2024-06-27 179 views
9

问题简述 想问一下能不能通过GPU训练呢(具体就是用synthesizer_train的时候能不能使用GPU?

复现与环境 我搜了一下,将synthesizer_train头部加上了 Le3P5d.png 请问下这样就可以使用GPU训练了么?

回答

4

已解决,问题在于pytorch是cpu版本,uninstall后重新下载即可

2

加了那个代码才能用GPU吗

9

是的,让你使用gpu而不是cpu