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的Django框架中的通用视图
May 04 Python
python Django批量导入数据
Mar 25 Python
Python人脸识别初探
Dec 21 Python
selenium+python 去除启动的黑色cmd窗口方法
May 22 Python
解决Tensorflow使用pip安装后没有model目录的问题
Jun 13 Python
Python 根据日志级别打印不同颜色的日志的方法示例
Aug 08 Python
Python socket模块方法实现详解
Nov 05 Python
pytorch AvgPool2d函数使用详解
Jan 03 Python
pandas读取csv文件提示不存在的解决方法及原因分析
Apr 21 Python
python使用tkinter实现透明窗体上绘制随机出现的小球(实例代码)
May 17 Python
Python趣味挑战之实现简易版音乐播放器
May 28 Python
python使用pycharm安装pyqt5以及相关配置
Apr 22 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的配置文件php.ini
2006/10/09 PHP
apache+codeigniter 通过.htcaccess做动态二级域名解析
2012/07/01 PHP
关于mysql字符集设置了character_set_client=binary 在gbk情况下会出现表描述是乱码的情况
2013/01/06 PHP
PHP实现模仿socket请求返回页面的方法
2014/11/04 PHP
在Mac OS上搭建PHP的Yii框架及相关测试环境
2016/02/14 PHP
PHP获取IP地址所在地信息的实例(使用纯真IP数据库qqwry.dat)
2016/11/15 PHP
php图形jpgraph操作实例分析
2017/02/22 PHP
原生JS实现Ajax通过POST方式与PHP进行交互的方法示例
2018/05/12 PHP
javascript基础的动画教程,直观易懂
2007/01/10 Javascript
JQUERY设置IFRAME的SRC值的代码
2010/11/30 Javascript
javascript禁制后退键(Backspace)实例代码
2013/11/15 Javascript
jQuery实现的动态伸缩导航菜单实例
2015/05/07 Javascript
jquery.cookie实现的客户端购物车操作实例
2015/12/24 Javascript
jQuery动态追加页面数据以及事件委托详解
2017/05/06 jQuery
微信小程序 侧滑删除(左滑删除)
2017/05/23 Javascript
详解用vue.js和laravel实现微信支付
2017/06/23 Javascript
快速搭建vue2.0+boostrap项目的方法
2018/04/09 Javascript
js 数组详细操作方法及解析合集
2018/06/01 Javascript
微信小程序使用for循环动态渲染页面操作示例
2018/12/25 Javascript
JavaScript实现星级评价效果
2019/05/17 Javascript
Python函数可变参数定义及其参数传递方式实例详解
2015/05/25 Python
Python OpenCV处理图像之图像像素点操作
2018/07/10 Python
利用python开发app实战的方法
2019/07/09 Python
django之自定义软删除Model的方法
2019/08/14 Python
Python cookie的保存与读取、SSL讲解
2020/02/17 Python
python GUI库图形界面开发之PyQt5访问系统剪切板QClipboard类详细使用方法与实例
2020/02/27 Python
Python reduce函数作用及实例解析
2020/05/08 Python
总结Pyinstaller的坑及终极解决方法(小结)
2020/09/21 Python
HTML5 FormData 方法介绍以及实现文件上传示例
2017/09/12 HTML / CSS
代理班主任的自我评价
2014/02/04 职场文书
歌颂祖国演讲稿
2014/05/04 职场文书
保护环境建议书400字
2014/05/13 职场文书
踏青活动策划方案
2014/08/19 职场文书
2014年关工委工作总结
2014/11/17 职场文书
诚信高考倡议书
2019/06/24 职场文书
MySQL 表空间碎片的概念及相关问题解决
2021/05/07 MySQL