编程 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系列之新版本导入httplib模块报ImportError解决方案

May 23 Python

在django中使用自定义标签实现分页功能

Jul 04 Python

python 全局变量的import机制介绍

Sep 07 Python

python中数据爬虫requests库使用方法详解

Feb 11 Python

CentOS 7下安装Python3.6 及遇到的问题小结

Nov 08 Python

python实现两个文件夹的同步

Aug 29 Python

Python3自定义json逐层解析器代码

May 11 Python

Django ORM filter() 的运用详解

May 14 Python

Python使用urlretrieve实现直接远程下载图片的示例代码

Aug 17 Python

pycharm进入时每次都是insert模式的解决方式

Feb 05 Python

我对PyTorch dataloader里的shuffle=True的理解

May 20 Python

Python3中最常用的5种线程锁实例总结

Jul 07 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 输出简单动态WAP页面

2009/06/09 PHP

PHP使用openssl扩展实现加解密方法示例

2020/02/20 PHP

JS 遮照层实现代码

2010/03/31 Javascript

获取css样式表内样式的js函数currentStyle(IE),defaultView(FF)

2011/02/14 Javascript

基于jQuery中对数组进行操作的方法

2013/04/16 Javascript

Web跨浏览器进程通信(Web跨域)

2013/04/17 Javascript

jquery DIV撑大让滚动条滚到最底部代码

2013/06/06 Javascript

用javascript添加控件自定义属性解析

2013/11/25 Javascript

jQuery对象的selector属性用法实例

2014/12/27 Javascript

jQuery form插件之ajaxForm()和ajaxSubmit()的可选参数项对象

2016/01/23 Javascript

JavaScript实现页面定时刷新（定时器，meta）

2016/10/12 Javascript

Vue过滤器的用法和自定义过滤器使用

2017/02/08 Javascript

jQuery获取单选按钮radio选中值与去除所有radio选中状态的方法

2017/05/20 jQuery

微信小程使用swiper组件实现图片轮播切换显示功能【附源码下载】

2017/12/12 Javascript

Spring boot 和Vue开发中CORS跨域问题解决

2018/09/05 Javascript

layui当点击文本框时弹出选择框,显示选择内容的例子

2019/09/02 Javascript

jQuery操作元素的内容和样式完整实例分析

2020/01/10 jQuery

javascript开发实现贪吃蛇游戏

2020/07/31 Javascript

解决removeEventListener 无法清除监听的问题

2020/10/30 Javascript

python excel使用xlutils类库实现追加写功能的方法

2018/05/02 Python

Python 最大概率法进行汉语切分的方法

2018/12/14 Python

Python日期时间Time模块实例详解

2019/04/15 Python

详解python websocket获取实时数据的几种常见链接方式

2019/07/01 Python

python错误调试及单元文档测试过程解析

2019/12/19 Python

pytorch中 gpu与gpu、gpu与cpu 在load时相互转化操作

2020/05/25 Python

PyTorch安装与基本使用详解

2020/08/31 Python

元旦文艺汇演主持词

2014/03/26 职场文书

物业项目经理岗位职责

2015/04/01 职场文书

2015年村计划生育工作总结

2015/04/28 职场文书

2015年驾驶员工作总结

2015/04/29 职场文书

党支部创先争优公开承诺书

2015/04/30 职场文书

公司的力量观后感

2015/06/05 职场文书

二十年同学聚会致辞

2015/07/28 职场文书

民间借贷纠纷起诉书

2015/08/03 职场文书

python实现的人脸识别打卡系统

2021/05/08 Python

python基础之文件操作

2021/10/24 Python