Pytorch 占用cpu资源过多

最近在用pytorch进行一些AI的工作时发现 其cpu占用量过高。

top

经过调试发现 主要是将数据从GPU下载到CPU上消耗资源比较高。

遇到这个问题，限制一下线程数量就可以了；

torch.set_num_threads(num)

这个方法改变CPU的占用量比较明显。