编程 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之模拟鼠标键盘动作具体实现

Dec 30 Python

python实现数组插入新元素的方法

May 22 Python

python中Apriori算法实现讲解

Dec 10 Python

对Python实现累加函数的方法详解

Jan 23 Python

在cmd中查看python的安装路径方法

Jul 03 Python

解决win7操作系统Python3.7.1安装后启动提示缺少.dll文件问题

Jul 15 Python

Python使用type关键字创建类步骤详解

Jul 23 Python

python找出列表中大于某个阈值的数据段示例

Nov 24 Python

Python-jenkins模块获取jobs的执行状态操作

May 12 Python

PyTorch 导数应用的使用教程

Aug 31 Python

用python对oracle进行简单性能测试

Dec 05 Python

Python数据可视化之Seaborn的安装及使用

Apr 19 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

Uncaught exception com_exception with message Failed to create COM object

2012/01/11 PHP

PHP入门教程之面向对象的特性分析(继承,多态,接口,抽象类,抽象方法等)

2016/09/11 PHP

PHP获取当前系统时间的方法小结

2018/10/03 PHP

Yii2框架自定义类统一处理url操作示例

2019/05/25 PHP

PHP上传图片到数据库并显示的实例代码

2019/12/20 PHP

PHP7 字符串处理机制修改

2021/03/09 PHP

JavaScript 快捷键设置实现代码

2009/03/13 Javascript

javascript window.open打开新窗口后无法再次打开该窗口问题的解决方法

2014/04/12 Javascript

单击某一段文字改写文本颜色

2014/06/06 Javascript

JavaScript sub方法入门实例（把字符串显示为下标）

2014/10/17 Javascript

JavaScript中的索引数组、关联数组和静态数组、动态数组讲解

2014/11/08 Javascript

script标签属性用type还是language

2015/01/21 Javascript

jQuery简单实现日历的方法

2015/05/04 Javascript

JS实现动态生成表格并提交表格数据向后端

2020/11/25 Javascript

一个字符串中出现次数最多的字符统计这个次数【实现代码】

2016/04/29 Javascript

基于javascript实现表格的简单操作

2016/05/21 Javascript

文本溢出插件jquery.dotdotdot.js使用方法详解

2017/06/22 jQuery

jQuery Collapse1.1.0折叠插件简单使用

2017/08/28 jQuery

JavaScript常用截取字符串的三种方式用法区别实例解析

2018/05/15 Javascript

element vue Array数组和Map对象的添加与删除操作

2018/11/14 Javascript

初学node.js中实现删除用户路由

2019/05/27 Javascript

原生JS实现天气预报

2020/06/16 Javascript

vue如何使用rem适配

2021/02/06 Vue.js

浅析python中numpy包中的argsort函数的使用

2018/08/30 Python

解决python给列表里添加字典时被最后一个覆盖的问题

2019/01/21 Python

Python中请不要再用re.compile了

2019/06/30 Python

Django后台admin的使用详解

2019/07/08 Python

PyTorch和Keras计算模型参数的例子

2020/01/02 Python

中国专业的综合网上购物商城：京东

2016/08/02 全球购物

韩国CJ食品专卖网：CJonmart

2016/09/11 全球购物

丹尼尔惠灵顿手表天猫官方旗舰店：Daniel Wellington

2017/08/25 全球购物

Napapijri西班牙在线商店：夹克、外套、运动衫等

2020/11/05 全球购物

在教室放鞭炮的检讨书

2014/09/28 职场文书

法定代表人资格证明书

2015/06/18 职场文书

校园歌手大赛主持词

2015/07/03 职场文书

Python初学者必备的文件读写指南

2021/06/23 Python