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 中国大陆许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步