报错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()