PyTorch RuntimeError: DataLoader worker (pid(s) 15804) exited unexpectedly

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

3、参考链接

  • RuntimeError: DataLoader worker (pid(s) 9528, 8320) exited unexpectedly

  • 【Python报错】RuntimeError: DataLoader worker (pid(s) 9764, 15128) exited unexpectedly

你可能感兴趣的:(PyTorch,深度学习)