pycharm上跑代码的时候,过不了多久代码就跑停报错,但是我显卡内存是充足的。
解决方案:
在报错的那一行代码前面加上两句话释放内存即可。
import torch, gc
gc.collect()
torch.cuda.empty_cache()
欢迎分享,转载请注明来源:内存溢出
pycharm上跑代码的时候,过不了多久代码就跑停报错,但是我显卡内存是充足的。
在报错的那一行代码前面加上两句话释放内存即可。
import torch, gc
gc.collect()
torch.cuda.empty_cache()
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)