在程序最开始的地方加上:
torch.multiprocessing.set_start_method(‘spawn’)
即可解决,如下第二行所示:
if __name__=='__main__':
torch.multiprocessing.set_start_method('spawn')
args = config_parser().parse_args()
data_loader = CreateDataLoader(args)
train()
其他详细的解释和解决方法:
- https://www.pythonheidong.com/blog/article/505157/8b5b28765cba2958ac02/
- https://www.pythonf.cn/read/65459