编程 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实现线程池的方法

Jun 30 Python

Python多线程、异步＋多进程爬虫实现代码

Feb 17 Python

python遍历 truple list dictionary的几种方法总结

Sep 11 Python

Python端口扫描简单程序

Nov 10 Python

浅析Python中yield关键词的作用与用法

Nov 29 Python

Python 包含汉字的文件读写之每行末尾加上特定字符

Dec 12 Python

python实现关键词提取的示例讲解

Apr 28 Python

python判断一个对象是否可迭代的例子

Jul 22 Python

Python调用Windows API函数编写录音机和音乐播放器功能

Jan 05 Python

python如何求数组连续最大和的示例代码

Feb 04 Python

Java多线程实现四种方式原理详解

Jun 02 Python

opencv实现图像几何变换

Mar 24 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中支持多种编码的中文字符串截取函数!

2007/03/20 PHP

php实现的双向队列类实例

2014/09/24 PHP

ThinkPHP实现分页功能

2017/04/28 PHP

PHP如何防止XSS攻击与XSS攻击原理的讲解

2019/03/22 PHP

得到文本框选中的文字，动态插入文字的js代码

2007/03/07 Javascript

JS DOM 操作实现代码

2010/08/01 Javascript

JavaScript中继承的一些示例方法与属性参考

2010/08/07 Javascript

jQuery实现文本展开收缩特效

2015/06/03 Javascript

Jquery实现顶部弹出框特效

2015/08/08 Javascript

jQuery点击其他地方时菜单消失的实现方法

2016/04/22 Javascript

angularjs ocLazyLoad分步加载js文件实例

2017/01/17 Javascript

jQuery实现拼图小游戏(实例讲解)

2017/07/24 jQuery

对vue里函数的调用顺序介绍

2018/03/17 Javascript

解决vue路由后界面没有变化,但是链接有的问题

2018/09/01 Javascript

vue开发环境配置跨域的方法步骤

2019/01/16 Javascript

解决vue cli使用typescript后打包巨慢的问题

2019/09/30 Javascript

[57:31]DOTA2-DPC中国联赛正赛 SAG vs CDEC BO3 第一场 2月1日

2021/03/11 DOTA

Python写的Socks5协议代理服务器

2014/08/06 Python

python打开网页和暂停实例

2014/09/30 Python

Python实现去除代码前行号的方法

2015/03/10 Python

深入解析Python中的__builtins__内建对象

2016/06/21 Python

Python下的Softmax回归函数的实现方法(推荐)

2017/01/26 Python

老生常谈python的私有公有属性(必看篇)

2017/06/09 Python

python将txt文件读入为np.array的方法

2018/10/30 Python

Django 路由控制的实现代码

2018/11/08 Python

python实现AES加密与解密

2019/03/28 Python

python+numpy实现的基本矩阵操作示例

2019/07/19 Python

OpenCV 表盘指针自动读数的示例代码

2020/04/10 Python

python中对二维列表中一维列表的调用方法

2020/06/07 Python

python实现不同数据库间数据同步功能

2021/02/25 Python

经济实惠的豪华背包和行李袋：Packs Project

2018/10/17 全球购物

CheapTickets香港机票预订网站：CheapTickets.hk

2019/06/26 全球购物

解释一下钝化(Swap out)

2016/12/26 面试题

乡镇党的群众路线教育实践活动领导班子对照检查材料

2014/09/25 职场文书

2016消防宣传标语口号

2015/12/26 职场文书

Java内存模型之happens-before概念详解

2021/06/13 Java/Android