Pytorch释放显存占用方式


Posted in Python onJanuary 13, 2020

如果在python内调用pytorch有可能显存和GPU占用不会被自动释放,此时需要加入如下代码

torch.cuda.empty_cache()

我们来看一下官方文档的说明

Releases all unoccupied cached memory currently held by the caching allocator so that those can be used in other GPU application and visible in nvidia-smi.

Note

empty_cache() doesn't increase the amount of GPU memory available for PyTorch. See Memory management for more details about GPU memory management.

此外还可以使用

memory_allocated()和max_memory_allocated()

观察显存占用,并使用

memory_cached()和 max_memory_cached()

观察由缓存分配器管理的内存。

以上这篇Pytorch释放显存占用方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python实现连接mongodb的方法
May 08 Python
通过数据库对Django进行删除字段和删除模型的操作
Jul 21 Python
Python书单 不将就
Jul 11 Python
Python KMeans聚类问题分析
Feb 23 Python
python实现数据导出到excel的示例--普通格式
May 03 Python
python实现图片批量压缩程序
Jul 23 Python
Pycharm无法使用已经安装Selenium的解决方法
Oct 13 Python
对Python3 pyc 文件的使用详解
Feb 16 Python
python基础教程之while循环
Aug 14 Python
使用Python进行中文繁简转换的实现代码
Oct 18 Python
python实现录屏功能(亲测好用)
Mar 02 Python
Python使用OpenCV实现虚拟缩放效果
Feb 28 Python
pytorch 实现删除tensor中的指定行列
Jan 13 #Python
python基于event实现线程间通信控制
Jan 13 #Python
pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
Jan 13 #Python
python、PyTorch图像读取与numpy转换实例
Jan 13 #Python
pytorch 彩色图像转灰度图像实例
Jan 13 #Python
Ranorex通过Python将报告发送到邮箱的方法
Jan 12 #Python
python opencv实现信用卡的数字识别
Jan 12 #Python
You might like
PHP个人网站架设连环讲(一)
2006/10/09 PHP
PHP读取目录下所有文件的代码
2008/01/07 PHP
PHP注释实例技巧
2008/10/03 PHP
php生成html文件方法总结
2014/12/01 PHP
php多重接口的实现方法
2015/06/20 PHP
PHP生成随机密码方法汇总
2015/08/27 PHP
详解php中 === 的使用
2016/10/24 PHP
PHP封装的验证码工具类定义与用法示例
2018/08/22 PHP
JavaScript入门教程(5) js Screen屏幕对象
2009/01/31 Javascript
使用JavaScript动态设置样式实现代码(2)
2013/01/25 Javascript
去掉gridPanel表头全选框的小例子
2013/07/18 Javascript
javascript面向对象快速入门实例
2015/01/13 Javascript
JavaScript比较两个对象是否相等的方法
2015/02/06 Javascript
jQuery插件bxSlider实现响应式焦点图
2015/04/12 Javascript
JavaScript中用getDate()方法返回指定日期的教程
2015/06/09 Javascript
详解Vue、element-ui、axios实现省市区三级联动
2019/05/07 Javascript
微信小程序实现商城倒计时
2020/11/01 Javascript
JS实现给数组对象排序的方法分析
2019/06/24 Javascript
javascript实现倒计时效果
2020/02/17 Javascript
python正则表达式match和search用法实例
2015/03/26 Python
Python实现的求解最大公约数算法示例
2018/05/03 Python
对python抓取需要登录网站数据的方法详解
2018/05/21 Python
python用for循环求和的方法总结
2019/07/08 Python
Numpy中ndim、shape、dtype、astype的用法详解
2020/06/14 Python
keras训练曲线,混淆矩阵,CNN层输出可视化实例
2020/06/15 Python
Python调用系统命令os.system()和os.popen()的实现
2020/12/31 Python
会计助理的岗位职责
2013/11/29 职场文书
七年级地理教学反思
2014/01/26 职场文书
青年志愿者活动总结
2014/04/26 职场文书
小学教师个人先进事迹材料
2014/05/17 职场文书
公司开业庆典策划方案
2014/06/04 职场文书
办公室领导干部作风整顿个人整改措施
2014/09/17 职场文书
学校捐书活动总结
2015/05/08 职场文书
2015年街道办事处工作总结
2015/05/22 职场文书
Python实现为PDF去除水印的示例代码
2022/04/03 Python
了解MySQL查询语句执行过程(5大组件)
2022/08/14 MySQL