pytorch num_worker>0 出错或者卡住不动了原因及解决方案
我的代码在data_loader中使用了opencv 来读取图片。
但是据说opencv3.4多线程是有bug的,会跟pytorch的多线程冲突。
num_worker>0 则代码崩溃,改成num_worker=0 则很慢.
解决方案:
import cv2
cv2.setNumThreads(0)
亲测解决!
我的代码在data_loader中使用了opencv 来读取图片。
但是据说opencv3.4多线程是有bug的,会跟pytorch的多线程冲突。
num_worker>0 则代码崩溃,改成num_worker=0 则很慢.
import cv2
cv2.setNumThreads(0)
亲测解决!