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