innnky / so-vits-svc

基于vits与softvc的歌声音色转换模型
GNU Affero General Public License v3.0
3.57k stars 1 forks source link

怎么设置cpu渲染 #146

Open leeseean opened 1 year ago

leeseean commented 1 year ago

由于显卡不行,所以想换成cpu渲染,该怎么设置?

SuCicada commented 1 year ago

不建议这么做,CPU效率比显卡慢了数个数量级。

Misaka17032 commented 1 year ago

一般来说,绝大部份的GPU的效率都高于CPU。但是如果显存不足的话,可以通过通过修改以下代码强行使用CPU推理:

将根目录下inference/infer_tool.py这个文件中的第153行self.dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")修改为self.dev = torch.device("cpu")