编程 Python

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释放显存占用方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持三水点靠木。

Pytorch释放显存占用方式

- Author -

冉冉升起的码农

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

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