从gpu中删除pytorch模型
从gpu中删除pytorch模型
提问于 2018-11-17 11:44:28
我想做一个交叉验证在我的项目的基础上，毕火炬。我也没有找到任何的方法，pytorch提供的删除当前的模型和空的GPU内存。你能告诉我怎么做吗？
回答 1
Stack Overflow用户
发布于 2018-11-17 16:32:31
释放PyTorch中的内存与正常的Python垃圾收集器一样工作。这意味着一旦所有对Python对象的引用都消失了，它就会被删除。
可以使用del运算符删除引用：
del model您必须确保没有对各自对象的引用，否则内存将不会被释放。
因此，一旦删除了model的所有引用，就应该删除它并释放内存。
如果您想了解更多关于内存管理的知识，可以在这里查看一下：https://pytorch.org/docs/stable/notes/cuda.html#cuda-memory-management
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/53350905
复制相关文章
相似问题
腾讯云开发者
