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

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

解决方案:

import cv2
cv2.setNumThreads(0)

亲测解决!

posted @ 2022-01-03 23:51  PengPanda  阅读(614)  评论(0编辑  收藏  举报