最近在用pytorch进行一些AI的工作时发现 其cpu占用量过高。
top
经过调试发现 主要是将数据从GPU下载到CPU上消耗资源比较高。
遇到这个问题,限制一下线程数量就可以了;
torch.set_num_threads(num)
这个方法改变CPU的占用量比较明显。

最近在用pytorch进行一些AI的工作时发现 其cpu占用量过高。
top
经过调试发现 主要是将数据从GPU下载到CPU上消耗资源比较高。
遇到这个问题,限制一下线程数量就可以了;
torch.set_num_threads(num)
这个方法改变CPU的占用量比较明显。