解决:RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB

素颜马尾好姑娘i 2023-10-04 17:01 134阅读 0赞

方法一:

仅需减小batchsize

改文件的配置cfg的batchsize=4,一般在cfg文件下的查找batch或batchsize,将batchsize调小后,再次运行,类似于改下面

在这里插入图片描述

方法二 :

上述方法还没解决,不改batchsize,可以考虑下面的方法的链接

不计算梯度:

ps: 在报错的哪一行代码的上面,加上下面一行代码,不计算梯度

with torch.no_grad()

不计算梯度的方法

方法三:

释放内存:链接如下

释放内存

  1. if hasattr(torch.cuda, 'empty_cache'):
  2. torch.cuda.empty_cache()

ps: 在报错的哪一行代码的上面,加上下面两行代码,释放无关的内存

  1. if hasattr(torch.cuda, 'empty_cache'):
  2. torch.cuda.empty_cache()

发表评论

表情:
评论列表 (有 0 条评论,134人围观)

还没有评论,来说两句吧...

相关阅读