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