训练模型的时候如果用的是 torch.nn.DataParallel 封装的模型 那么请设置多GPU CUDA = '0,1' os.environ['CUDA_VISIBLE_DEVICES'] = CUDA