RuntimeError: Cannot re-initialize CUDA in forked subprocess解决方法之一

在程序最开始的地方加上:

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

你可能感兴趣的:(#,Code,pytorch,CUDA,subprocess)