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在Matplotlib图中显示中文字体的操作方法
这篇文章主要介绍了Python在Matplotlib图中显示中文字体的方法,本篇主要针对在Ubuntu系统中,matplotlib显示不了中文的问题,尤其是在无法安装系统字体的情况下,解决Python绘图时中文显示的问题。需要的朋友可以参考下2019-07-07
python使用pika库调用rabbitmq交换机模式详解
这篇文章主要介绍了python使用pika库调用rabbitmq交换机模式详解,文章围绕主题展开详细的内容介绍,具有一定的参考价值,感兴趣的小伙伴可以参考一下2022-08-08
Python实现将16进制字符串转化为ascii字符的方法分析
这篇文章主要介绍了Python实现将16进制字符串转化为ascii字符的方法,结合实例形式分析了Python 16进制字符串转换为ascii字符的实现方法与相关注意事项,需要的朋友可以参考下2017-07-07
带你彻底搞懂python操作mysql数据库(cursor游标讲解)
这篇文章主要介绍了带你彻底搞懂python操作mysql数据库(cursor游标讲解),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2020-01-01


最新评论