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)