pytorch num_worker>0 出错或者卡住不动了原因及解决方案

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

解决方案:

import cv2
cv2.setNumThreads(0)

亲测解决!

本文作者:PengPanda

本文链接:https://www.cnblogs.com/pengpanda/p/15760992.html

版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。

posted @   PengPanda  阅读(829)  评论(0编辑  收藏  举报
点击右上角即可分享
微信分享提示
💬
评论
📌
收藏
💗
关注
👍
推荐
🚀
回顶
收起