gpu - pytorch 从 gpu 中删除模型

Question

我想在基于 Pytorch 的项目中进行交叉验证。
而且我也没有找到pytorch提供的删除当前模型和清空GPU内存的方法。你能告诉我该怎么做吗？

Answer 1

在 PyTorch 中释放内存的工作方式与普通 Python 垃圾收集器一样。这意味着一旦对 Python 对象的所有引用都消失了，它将被删除。

您可以使用 del 删除引用。运算符(operator):

del model

你必须确保没有对相应对象的引用，否则内存不会被释放。

因此，一旦您删除了 model 的所有引用，它应该被删除并释放内存。

如果您想了解有关内存管理的更多信息，可以查看此处:
https://pytorch.org/docs/stable/notes/cuda.html#cuda-memory-management