当前位置: 代码迷 >> 综合 >> 测试时报错 RuntimeError: CUDA out of memory.
  详细解决方案

测试时报错 RuntimeError: CUDA out of memory.

热度:100   发布时间:2023-09-28 22:14:47.0

问题描述:

训练神经网络模型,训练时正常,训练一个epoch后测试的时候,报错RuntimeError: CUDA out of memory. 然而我的模型并没有很大,GPU占用率也不高。

解决方法:

with torch.no_grad():(your test code)

 

 

  相关解决方案