Open leeseean opened 1 year ago
不建议这么做,CPU效率比显卡慢了数个数量级。
一般来说,绝大部份的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")
由于显卡不行,所以想换成cpu渲染,该怎么设置?