我的问题是,当我运行FC网络时,代码在CPU和GPU中都能正常工作。但是,当涉及到CNN时,我只能在CPU上进行训练。当我尝试在GPU上进行训练时,会引发错误。
喜欢:
以退出代码-1073741819(0xC0000005)完成的过程
我发现当代码转到loss.backword时引发的错误。当我使用第一列而不是第二列时发生了错误。
device = torch.device("cuda:0")
device = torch.device("cuda:0" if opt.cuda else "cpu")
我的环境是Python 3.6.9,Windows 10,Torch 1.2.0,Cuda 9.2。
答案 0 :(得分:1)
最后,我弄清楚了。
发生此错误仅仅是因为我的一个变量未加载到cuda中。
当我添加此输出= Variable(netD(real_cpu),requires_grad=True)
时,问题解决了。