RuntimeError: CUDA error: an illegal memory access
时间: 2023-11-03 10:55:07 浏览: 230
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
5星 · 资源好评率100%
根据引用和引用的内容,出现RuntimeError: CUDA error: an illegal memory access错误的原因可能有多种解况。首先,可以尝试减小batch_size、将所有网络和输入放入相同设备的net的train和eval状态、检查是否有防止内存混乱的代码。另外,可以尝试设置CUDA_LAUNCH_BLOCKING=1和torch.cuda.set_device(1)来解决问题。还有可能是由于cudnn版本不对,需要重新安装。最后,根据引用的建议,回退pytorch版本到v1.11.0可能也能解决这个问题。你可以尝试以下操作,逐一排查原因并找到合适的解决办法来解决该错误。<span class="em">1</span><span class="em">2</span>
#### 引用[.reference_title]
- *1* *2* [CUDA RuntimeError: CUDA error: an illegal memory access was encountered](https://blog.csdn.net/u010590593/article/details/127410358)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
阅读全文