@创建于: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’内。