python - 如何修复错误“进程以退出代码-1073741819(0xC0000005)完成”
问题描述
我的问题是当我运行 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。
解决方案
最后,我想通了。
发生此错误只是因为我的一个变量未在 cuda 中加载。
当我添加这个输出=Variable(netD(real_cpu),requires_grad=True)
问题解决了。