编程 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使用struct处理二进制的实例详解

Sep 11 Python

Python利用lxml模块爬取豆瓣读书排行榜的方法与分析

Apr 15 Python

Python数据结构与算法（几种排序）小结

Jun 22 Python

基于python-opencv3的图像显示和保存操作

Jun 27 Python

解决python多行注释引发缩进错误的问题

Aug 23 Python

python3获取url文件大小示例代码

Sep 18 Python

python实现多进程按序号批量修改文件名的方法示例

Dec 30 Python

Django中ORM找出内容不为空的数据实例

May 20 Python

Python如何安装第三方模块

May 28 Python

详解基于Scrapy的IP代理池搭建

Sep 29 Python

Django数据库迁移常见使用方法

Nov 12 Python

Python+tkinter实现高清图片保存

Mar 13 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

解决PhpMyAdmin中导入2M以上大文件限制的方法分享

2014/06/06 PHP

PHP中使用数组指针函数操作数组示例

2014/11/19 PHP

Laravel实现搜索的时候分页并携带参数

2019/10/15 PHP

服务端 VBScript 与 JScript 几个相同特性的写法 By shawl.qiu

2007/03/06 Javascript

用JavaScript页面不刷新时全选择，全删除（GridView）

2009/04/14 Javascript

精通JavaScript 纠正 cleanWhitespace函数

2010/03/11 Javascript

jquery中event对象属性与方法小结

2013/12/18 Javascript

jquery学习总结（超级详细）

2014/09/04 Javascript

thinkphp 表名大小写窍门

2015/02/01 Javascript

jQuery中delegate()方法的用法详解

2016/10/13 Javascript

jQuery插件fullPage.js实现全屏滚动效果

2016/12/02 Javascript

webpack踩坑之路图片的路径与打包

2017/09/05 Javascript

使用async、enterproxy控制并发数量的方法详解

2018/01/02 Javascript

nodejs+mongodb+vue前后台配置ueditor的示例代码

2018/01/02 NodeJs

教你如何编写Vue.js的单元测试的方法

2018/10/17 Javascript

Vue批量图片显示时遇到的路径被解析问题

2019/03/28 Javascript

Vuejs通过拖动改变元素宽度实现自适应

2020/09/02 Javascript

JavaScript 实现继承的几种方式

2021/02/19 Javascript

[04:50]2019DOTA2高校联赛秋季赛四强集锦

2019/12/27 DOTA

利用Python爬虫给孩子起个好名字

2017/02/14 Python

python 文本单词提取和词频统计的实例

2018/12/22 Python

python 并发编程阻塞IO模型原理解析

2019/08/20 Python

python实现代码统计程序

2019/09/19 Python

win10下python2和python3共存问题解决方法

2019/12/23 Python

pytorch 实现在一个优化器中设置多个网络参数的例子

2020/02/20 Python

利用Python实现斐波那契数列的方法实例

2020/07/26 Python

html5+css3实现一款注册表单实例

2013/04/17 HTML / CSS

美国探亲签证邀请信

2014/02/05 职场文书

养牛场项目建议书

2014/05/13 职场文书

工作经历证明书范文

2014/11/02 职场文书

优秀高中学生评语

2014/12/30 职场文书

市场部岗位职责

2015/02/12 职场文书

2015年行政工作总结范文

2015/04/09 职场文书

2015年环保局工作总结

2015/05/22 职场文书

Pytorch 统计模型参数量的操作 param.numel()

2021/05/13 Python

简单谈谈Python面向对象的相关知识

2021/06/28 Python