码迷,mamicode.com
首页 > 其他好文 > 详细

pytorch占用过多CPU问题

时间:2019-12-02 11:42:40      阅读:396      评论:0      收藏:0      [点我收藏+]

标签:比较   color   style   bsp   tor   pytorch   方法   export   exp   

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

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

法二、export OMP_NUM_THREADS = 1 (未测)

pytorch占用过多CPU问题

标签:比较   color   style   bsp   tor   pytorch   方法   export   exp   

原文地址:https://www.cnblogs.com/zf-blog/p/11969615.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!