pytorch程序异常后删除占用的显存操作
更新时间：2020年01月13日 09:56:51 作者：upDiff
今天小编就为大家分享一篇pytorch程序异常后删除占用的显存操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
1-删除模型变量
del model_define
2-清空CUDA cache
torch.cuda.empty_cache()
3-步骤2（异步）需要一定时间，设置时延
time.sleep(5)
完整代码如下：
del styler torch.cuda.empty_cache() time.sleep(5)
以上这篇pytorch程序异常后删除占用的显存操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家介绍了Python 类中定义多个构造器方法重载与泛方法示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-03-03
python下的ip正则实现代码。2009-05-05
这篇文章主要介绍了Python排序搜索基本算法之插入排序,结合实例形式分析了基于比较的插入排序和基于交换的插入排序实现技巧,需要的朋友可以参考下2017-12-12
itchat和matplotlib的结合使用爬取微信信息的实例
下面小编就为大家带来一篇itchat和matplotlib的结合使用爬取微信信息的实例。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-08-08
这篇文章主要介绍了Python 找出出现次数超过数组长度一半的元素实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
这篇文章主要介绍了Django查询优化及ajax编码格式原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-03-03
这篇文章主要介绍了python如何修改PYTHONPATH环境变量问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-08-08
如何使用python-opencv批量生成带噪点噪线的数字验证码
这篇文章主要介绍了如何使用python-opencv批量生成带噪点噪线的数字验证码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-12-12
这篇文章主要介绍了django框架中间件原理与用法,结合实例形式详细分析了Django框架常用中间件与基本使用技巧,需要的朋友可以参考下2019-12-12
这篇文章主要为大家介绍了Python使用虚拟环境(安装下载更新卸载)命令,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-11-11
最新评论