报错runtimerror:cuda out of memory.tried to allocate 98.00MB(GPU 0;10.91 GiB total capacity;)：
pycharm上跑代码的时候，过不了多久代码就跑停报错，但是我显卡内存是充足的。
解决方案：
在报错的那一行代码前面加上两句话释放内存即可。
import torch, gc
gc.collect()
torch.cuda.empty_cache()
文章出处登录后可见！
已经登录？立即刷新
报错runtimerror:cuda out of memory.tried to allocate 98.00MB(GPU 0；10.91 GiB total capacity；)
赞 (0)
Command ‘pip‘ not found, but can be installed with:
上一篇
2022年5月10日