RuntimeError: CUDA error: an illegal memory access was encountered 错误解决方案

RuntimeError: CUDA error: an illegal memory access was encountered

首先，大家先检查自己的网络的参数是否有问题，如果参数有问题会导致此问题。

其次，博主遇到一个情况。在单GPU下开启时，eval阶段会报这种错误。

torch.nn.DataParallel(net,device_ids=[0])

在net eval之前加上这句话，错误就没有了。现在也不知道是什么原因，但是可以看出增加或删除这句话，会导致错误的消失与出现。

with torch.no_grad():
     out = net.eval(your param)