脚本宝典收集整理的这篇文章主要介绍了pytorch num_worker>0 出错或者卡住不动了原因及解决方案，脚本宝典觉得挺不错的，现在分享给大家，也给大家做个参考。

我的代码在data_loader中使用了opencv 来读取图片。 但是据说opencv3.4多线程是有bug的，会跟pytorch的多线程冲突。 num_worker>0 则代码崩溃，改成num_worker=0 则很慢.

解决方案：

import cv2
cv2.setNumThreads(0)

亲测解决！

脚本宝典总结

以上是脚本宝典为你收集整理的pytorch num_worker>0 出错或者卡住不动了原因及解决方案全部内容，希望文章能够帮你解决pytorch num_worker>0 出错或者卡住不动了原因及解决方案所遇到的问题。

如果觉得脚本宝典网站内容还不错，欢迎将脚本宝典推荐好友。

本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ：384754419，请注明来意。