-
解决运行pytorch程序多线程问题
- 本机window10
![](https://images2015.cnblogs.com/blog/871133/201706/871133-20170630105408493-154027012.jpg)
- linux服务器开启了多个进程
![](https://images2015.cnblogs.com/blog/871133/201706/871133-20170630105451055-1708362290.jpg)
- Linux服务器占用大量CPU
![](https://images2015.cnblogs.com/blog/871133/201706/871133-20170630105614868-1198158471.jpg)
- 在pytorch中提供了(http://pytorch.org/docs/master/multiprocessing.html)muliprocessing多进程管理,其中有Pool进程池、Process()等对进程进行管理,可能是我使用的方式不对,反正是没有什么效果。
- 然而最简单的也是最好的解决办法 :torch.set_num_threads(int thread) ,可以很好的解决windows问题,参考(http://pytorch.org/docs/master/torch.html#parallelism) 然而,在linux服务器上还是有一些问题的,export OMP_NUM_THREADS = 1 可以解决Linux问题。
-
相关阅读:
迪杰斯特拉算法
基数排序
快排算法
插入排序与希尔排序算法
java--jmm知识
Java基础
socket代理
TestLink 学习第一周
软件体系结构第三章之解释器风格
如何衡量个人在各自团队的效率和绩效
-
原文地址:https://www.cnblogs.com/bamtercelboo/p/7097933.html
Copyright © 2020-2023
润新知