编程 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增量循环删除MySQL表数据的方法

Sep 23 Python

python实现自动发送报警监控邮件

Jun 21 Python

基于Python开发chrome插件的方法分析

Jul 07 Python

python之文件读取一行一行的方法

Jul 12 Python

python爬虫之urllib,伪装,超时设置,异常处理的方法

Dec 19 Python

python图形工具turtle绘制国际象棋棋盘

May 23 Python

pandas的连接函数concat()函数的具体使用方法

Jul 09 Python

python 叠加等边三角形的绘制的实现

Aug 14 Python

python Django框架实现web端分页呈现数据

Oct 31 Python

Python中使用threading.Event协调线程的运行详解

May 02 Python

python实战之一步一步教你绘制小猪佩奇

Apr 22 Python

详解python字符串驻留技术

May 21 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

解析link_mysql的php版

2013/06/30 PHP

微信接口生成带参数的二维码

2017/07/31 PHP

PHP 枚举类型的管理与设计知识点总结

2020/02/13 PHP

javascript之水平横向滚动歌词同步的应用

2007/05/07 Javascript

javascript 框架小结个人工作经验

2009/06/13 Javascript

用JS提交参数创建form表单在FireFox中遇到的问题

2013/01/16 Javascript

jquery检测input checked 控件是否被选中的方法

2014/03/26 Javascript

js实现背景图片感应鼠标变化的方法

2015/02/28 Javascript

举例讲解Node.js中的Writable对象

2015/07/29 Javascript

js实现带缓冲效果的仿QQ面板折叠菜单代码

2015/09/06 Javascript

js实现select二级联动下拉菜单

2020/04/17 Javascript

微信小程序使用第三方库Immutable.js实例详解

2016/09/27 Javascript

JS中获取 DOM 元素的绝对位置实例详解

2018/04/23 Javascript

原生JS实现的轮播图功能详解

2018/08/06 Javascript

解决Layui选择全部,换页checkbox复选框重新勾选的问题方法

2018/08/14 Javascript

Mint UI实现A-Z字母排序的城市选择列表

2018/12/28 Javascript

Javascript实现一朵从含苞到绽放的玫瑰

2019/03/30 Javascript

addEventListener()和removeEventListener()追加事件和删除追加事件

2020/12/04 Javascript

[00:37]2016完美“圣”典风云人物：AMS宣传片

2016/12/06 DOTA

Python图像灰度变换及图像数组操作

2016/01/27 Python

Python爬取网易云音乐热门评论

2017/03/31 Python

深入理解Django的自定义过滤器

2017/10/17 Python

Python 调用 Outlook 发送邮件过程解析

2019/08/08 Python

python matplotlib库直方图绘制详解

2019/08/10 Python

Python tkinter实现简单加法计算器代码实例

2020/05/13 Python

amazeui页面校验功能的实现代码

2020/08/24 HTML / CSS

MONNIER Frères英国官网：源自巴黎女士奢侈品配饰电商平台

2018/12/06 全球购物

软件毕业生个人鉴定

2014/03/03 职场文书

高中军训感言600字

2014/03/11 职场文书

党员评议思想汇报

2014/10/08 职场文书

2015社区六五普法工作总结

2015/04/21 职场文书

2015年出纳工作总结与计划

2015/05/18 职场文书

幼儿园开学家长寄语（2015秋季）

2015/05/27 职场文书

六一文艺汇演主持词

2015/06/30 职场文书

基于Python绘制子图及子图刻度的变换等的问题

2021/05/23 Python

Go语言特点及基本数据类型使用详解

2022/03/21 Golang