pytorch占用过多CPU问题

Linux下,使用pytorch有时候会出现占用过多CPU资源的问题(占用过多线程),解决方法如下:

法一、torch.set_num_threads(int thread) (亲测比较有效)

法二、export OMP_NUM_THREADS = 1 (未测)

posted @ 2019-12-02 11:29  outthinker  阅读(4803)  评论(0编辑  收藏  举报