编程 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模拟登陆Tom邮箱示例分享

Jan 13 Python

Python功能键的读取方法

May 28 Python

详解Django缓存处理中Vary头部的使用

Jul 24 Python

Python3中在Anaconda环境下安装basemap包

Oct 21 Python

Python产生Gnuplot绘图数据的方法

Nov 09 Python

浅谈pyqt5在QMainWindow中布局的问题

Jun 21 Python

详解python编译器和解释器的区别

Jun 24 Python

详解用pyecharts Geo实现动态数据热力图城市找不到问题解决

Jun 26 Python

python实现发送form-data数据的方法详解

Sep 27 Python

Python测试线程应用程序过程解析

Dec 31 Python

Python re正则表达式元字符分组()用法分享

Feb 10 Python

PyTorch安装与基本使用详解

Aug 31 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遍历目录与文件夹的多种方法详解

2013/11/14 PHP

Web程序工作原理详解

2014/12/25 PHP

各种快递查询--Api接口

2016/04/26 PHP

功能强大的php文件上传类

2016/08/29 PHP

JS中字符问题(二进制/十进制/十六进制及ASCII码之间的转换)

2008/11/03 Javascript

var与Javascript变量隐式声明

2009/09/17 Javascript

jquery中的mouseleave和mouseout的区别模仿下拉框效果

2012/02/07 Javascript

javascript变量作用域使用中常见错误总结

2013/03/26 Javascript

jQuery实现点击小图显示大图代码分享

2015/08/25 Javascript

JQuery通过AJAX从后台获取信息显示在表格上并支持行选中

2015/09/15 Javascript

JQuery实现的按钮倒计时效果

2015/12/23 Javascript

jQuery实现元素拖拽并cookie保存顺序的方法

2016/02/20 Javascript

js表单验证实例讲解

2016/03/31 Javascript

jQuery的ajax下载blob文件

2016/07/21 Javascript

JS克隆，属性，数组，对象，函数实例分析

2016/11/26 Javascript

详解webpack分包及异步加载套路

2017/06/29 Javascript

jQuery实现模糊搜索功能的方法分析

2018/06/29 jQuery

JavaScript变速动画函数封装添加任意多个属性

2019/04/03 Javascript

vue学习之Vue-Router用法实例分析

2020/01/06 Javascript

小程序中使用css var变量(使js可以动态设置css样式属性)

2020/03/31 Javascript

全面解析js中的原型,原型对象,原型链

2021/01/25 Javascript

初学Python实用技巧两则

2014/08/29 Python

python模块之StringIO使用示例

2015/04/08 Python

Python实现计算两个时间之间相差天数的方法

2017/05/10 Python

Python3使用正则表达式爬取内涵段子示例

2018/04/22 Python

详解配置Django的Celery异步之路踩坑

2018/11/25 Python

详解pyenv下使用python matplotlib模块的问题解决

2018/11/29 Python

python 使用 requests 模块发送http请求的方法

2018/12/09 Python

泰国网上购物：Shopee泰国

2018/09/14 全球购物

后勤自我鉴定

2013/10/13 职场文书

大家检讨书5000字

2014/02/03 职场文书

服装促销活动方案

2014/02/23 职场文书

信息技术课后反思

2014/04/27 职场文书

初中毕业典礼演讲稿

2014/09/09 职场文书

2015年置业顾问工作总结

2015/04/07 职场文书

JavaScript中的LHS和RHS分析详情

2022/04/06 Javascript