编程 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的差集、交集与并集的方法

Nov 01 Python

使用Python编写一个在Linux下实现截图分享的脚本的教程

Apr 24 Python

python+selenium实现京东自动登录及秒杀功能

Nov 18 Python

浅析Python3爬虫登录模拟

Feb 07 Python

python实现可逆简单的加密算法

Mar 22 Python

Pandas中resample方法详解

Jul 02 Python

python的json中方法及jsonpath模块用法分析

Dec 06 Python

python如何使用jt400.jar包代码实例

Dec 20 Python

Python 实现将数组/矩阵转换成Image类

Jan 09 Python

如何对python的字典进行排序

Jun 19 Python

Scrapy模拟登录赶集网的实现代码

Jul 07 Python

python自然语言处理之字典树知识总结

Apr 25 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 XML error parsing SOAP payload on line 1

2010/06/17 PHP

ThinkPHP模板Switch标签用法示例

2014/06/30 PHP

PHP CodeIgniter框架的工作原理研究

2015/03/30 PHP

android上传图片到PHP的过程详解

2015/08/03 PHP

[原创]PHP正则删除html代码中a标签并保留标签内容的方法

2017/05/23 PHP

解读JavaScript代码 var ie = !-[1,] 最短的IE判定代码

2011/05/28 Javascript

jquery中EasyUI使用技巧小结

2015/02/10 Javascript

javascript比较两个日期相差天数的方法

2015/07/23 Javascript

Bootstrap进度条组件知识详解

2016/05/01 Javascript

详解用webpack2搭建angular2的项目

2017/06/22 Javascript

关于vue中watch检测到不到对象属性的变化的解决方法

2018/02/08 Javascript

详解如何运行vue项目

2019/04/15 Javascript

js实现图片无缝循环轮播

2019/10/28 Javascript

vue自定义正在加载动画的例子

2019/11/14 Javascript

微信小程序后端实现授权登录

2020/02/24 Javascript

利用node.js开发cli的完整步骤

2020/12/29 Javascript

[59:36]2018DOTA2亚洲邀请赛 4.3 突围赛 Secret vs VG 第二场

2018/04/04 DOTA

[11:12]2018DOTA2国际邀请赛寻真——绿色长城OpTic

2018/08/10 DOTA

Python学习小技巧之利用字典的默认行为

2017/05/20 Python

Python numpy生成矩阵、串联矩阵代码分享

2017/12/04 Python

django2+uwsgi+nginx上线部署到服务器Ubuntu16.04

2018/06/26 Python

PYTHON EVAL的用法及注意事项解析

2019/09/06 Python

浅析PEP570新语法: 只接受位置参数

2019/10/15 Python

python模拟点击在ios中实现的实例讲解

2020/11/26 Python

利用纯css3实现的文字亮光特效的代码演示

2014/11/27 HTML / CSS

高级护理专业毕业生推荐信

2013/12/25 职场文书

投资意向书范本

2014/04/01 职场文书

师德演讲稿范文

2014/05/06 职场文书

锦旗标语大全

2014/06/23 职场文书

党支部三严三实对照检查材料思想汇报

2014/09/29 职场文书

2014年机关党委工作总结

2014/12/11 职场文书

钱塘江大潮导游词

2015/02/03 职场文书

物业工程部岗位职责

2015/02/11 职场文书

运动会200米广播稿

2015/08/19 职场文书

redis缓存存储Session原理机制

2021/11/20 Redis

js前端设计模式优化50%表单校验代码示例

2022/06/21 Javascript