编程 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之有容乃大的list(4)

Sep 28 Python

Python文档生成工具pydoc使用介绍

Jun 02 Python

Python文件读取的3种方法及路径转义

Jun 21 Python

python字符串中的单双引

Feb 16 Python

python编程使用selenium模拟登陆淘宝实例代码

Jan 25 Python

解决tensorflow测试模型时NotFoundError错误的问题

Jul 27 Python

Python 实现两个列表里元素对应相乘的方法

Nov 14 Python

django多对多表的创建，级联删除及手动创建第三张表

Jul 25 Python

Python 实用技巧之利用Shell通配符做字符串匹配

Aug 23 Python

python内置模块collections知识点总结

Dec 19 Python

Python流程控制语句的深入讲解

Jun 15 Python

pycharm部署django项目到云服务器的详细流程

Jun 29 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/08/07 PHP

PHP DB 数据库连接类定义与用法示例

2019/03/11 PHP

自己动手写的javascript前端等待控件

2015/10/30 Javascript

js+css简单实现网页换肤效果

2015/12/29 Javascript

javascript实现checkbox复选框实例代码

2016/01/10 Javascript

JavaScript实现获取某个元素相邻兄弟节点的prev与next方法

2016/01/25 Javascript

js判断鼠标位置是否在某个div中的方法

2016/02/26 Javascript

JS判断元素是否在数组内的实现代码

2016/03/30 Javascript

第二章之Bootstrap 页面排版样式

2016/04/25 Javascript

原生JS:Date对象全面解析

2016/09/06 Javascript

jQuery元素属性操作实例(设置、获取及删除元素属性)

2016/09/08 Javascript

pc加载更多功能和移动端下拉刷新加载数据

2016/11/07 Javascript

vue.js指令和组件详细介绍及实例

2017/04/06 Javascript

javascript 中关于array的常用方法详解

2017/05/05 Javascript

js插件实现图片滑动验证码

2020/09/29 Javascript

JavaScript设计模式之装饰者模式实例详解

2019/01/17 Javascript

vue实现点击按钮切换背景颜色的示例代码

2020/06/23 Javascript

[54:47]Liquid vs VP Supermajor决赛 BO 第五场 6.10

2018/07/05 DOTA

[52:37]完美世界DOTA2联赛循环赛 Forest vs DM BO2第一场 10.29

2020/10/29 DOTA

Python简明入门教程

2015/08/04 Python

使用Django连接Mysql数据库步骤

2019/01/15 Python

anaconda中更改python版本的方法步骤

2019/07/14 Python

python 两个数据库postgresql对比

2019/10/21 Python

利用python对excel中一列的时间数据更改格式操作

2020/07/14 Python

使用Python制作一个数据预处理小工具(多种操作一键完成)

2021/02/07 Python

英国信箱在线鲜花速递公司：Bloom & Wild

2019/03/10 全球购物

.NET常见笔试题集

2012/12/01 面试题

房地产还款计划书

2014/01/10 职场文书

会计专业自我评价

2014/02/12 职场文书

教师自我反思材料

2014/02/14 职场文书

《阳光》教学反思

2014/02/23 职场文书

银行先进个人事迹材料

2014/05/11 职场文书

党员证明信

2015/06/19 职场文书

大学生社会服务心得体会

2016/01/22 职场文书

为什么mysql字段要使用NOT NULL

2021/05/13 MySQL

Docker下安装Oracle19c

2022/04/13 Servers