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中RegEx在数据处理中的应用
正则表达式（Regular Expressions,简称 RegEx）是一种强大的文本匹配和搜索工具,它在数据处理、文本解析和字符串操作中发挥着关键作用,下面就跟随小编一起来了解一下RegEx的具体使用吧
2024-01-01
用selenium解决滑块验证码的实现步骤
验证码作为一种自然人的机器人的判别工具,被广泛的用于各种防止程序做自动化的场景中,下面这篇文章主要给大家介绍了关于用selenium解决滑块验证码的实现步骤,需要的朋友可以参考下
2023-02-02
Python增强赋值和共享引用注意事项小结
这篇文章主要给大家介绍了关于Python增强赋值和共享引用注意事项的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
2019-05-05
Python数据结构与算法之图的基本实现及迭代器实例详解
这篇文章主要介绍了Python数据结构与算法之图的基本实现及迭代器,结合实例形式详细分析了数据结构与算法中图的实现及迭代器相关算法原理与操作技巧,需要的朋友可以参考下
2017-12-12
Apache,wsgi,django 程序部署配置方法详解
这篇文章主要介绍了Apache,wsgi,django 程序部署配置方法,结合实例形式详细分析了Linux环境下Apache,wsgi,django程序部署配置的相关操作技巧与注意事项,需要的朋友可以参考下
2019-07-07
python opencv根据颜色进行目标检测的方法示例
这篇文章主要介绍了python opencv根据颜色进行目标检测的方法示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-01-01
python表格存取的方法
这篇文章主要为大家详细介绍了python表格存取的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-03-03
python画图——实现在图上标注上具体数值的方法
今天小编就为大家分享一篇python画图——实现在图上标注上具体数值的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-07-07
Python Django 命名空间模式的实现
这篇文章主要介绍了Python Django 命名空间模式的实现,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-08-08
python linecache 处理固定格式文本数据的方法
今天小编就为大家分享一篇python linecache 处理固定格式文本数据的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-01-01