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之前加...
·
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)
「智能机器人开发者大赛」官方平台，致力于为开发者和参赛选手提供赛事技术指导、行业标准解读及团队实战案例解析;聚焦智能机器人开发全栈技术闭环，助力开发者攻克技术瓶颈，促进软硬件集成、场景应用及商业化落地的深度研讨。 加入智能机器人开发者社区iRobot Developer，与全球极客并肩突破技术边界，定义机器人开发的未来范式！
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)