【pytorch】随着epoch增加，显存逐渐增加？解决pytorch显存释放问题_epoch完成后部分gpu显存占用变大

训练时

在训练时，检测 “out of memory” 的error并通过torch.cuda.empty_cache()处理
如：

      try:
          outputs = net(inputs)
      except RuntimeError as exception:
          if "out of memory" in str(exception):
              print('WARNING: out of memory, will pass this')     
              torch.cuda.empty_cache()
              continue
          else:
              raise exception
1
2
3
4
5
6
7
8
9

测试时

在测试时，避免忘记设置 torch.no_grad()
如：

with torch.no_grad():
    inputs = None
    outputs = model(inputs)
1
2
3

参考文献

Pytroch - 显存释放问题