@创建于:2021.10.20
1、报错
RuntimeError: DataLoader worker (pid(s) 15804) exited unexpectedly
网络上给出了方案,如:
- cuda 虚拟环境的共享内存不足,增大内存;
- 改成更小的batchsize。
2、我的方案
我的代码中,num_workers = 2,启用了多进程。
Python中多进程(multiprocessing这个模块包)的内容必须放在if name == 'main’之内才可以。多线程(threading这个模块包)是没有这项限定的。
所以,如果使用多进程,请务必放在if name == 'main’内。
本文探讨了如何修复RuntimeError: DataLoaderworker exited unexpectedly的问题,重点在于Python多进程设置和可能的原因,如共享内存不足和批大小调整。作者建议将多进程代码置于if __name__ == '__main__'内,并提供了相关解决方案和参考链接。

9560

被折叠的 条评论
为什么被折叠?



