pytorch占用过多CPU问题
Posted zf-blog
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch占用过多CPU问题相关的知识，希望对你有一定的参考价值。
Linux下，使用pytorch有时候会出现占用过多CPU资源的问题（占用过多线程），解决方法如下：
法一、torch.set_num_threads(int thread) （亲测比较有效）
法二、export OMP_NUM_THREADS = 1 （未测）
以上是关于pytorch占用过多CPU问题的主要内容，如果未能解决你的问题，请参考以下文章