在程序最开始的地方加上：
torch.multiprocessing.set_start_method(‘spawn’)torch.multiprocessing
阅读背景：
RuntimeError: Cannot re-initialize CUDA in forked subprocess解决方法之一_NLOS的博客
来源：互联网
分享到：
