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)
亲测解决!