Pytorch释放显存占用方式
更新时间：2020年01月13日 09:25:04 作者：冉冉升起的码农
今天小编就为大家分享一篇Pytorch释放显存占用方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
如果在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释放显存占用方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
当在图像上训练深度神经网络模型时，通过对由数据增强生成的更多图像进行训练，可以使模型更好地泛化。本文将为大家介绍Python基于域相关的图像增强实现方法，需要的可以了解一下2022-01-01
所谓决策树，就是一个类似于流程图的树形结构，树内部的每一个节点代表的是对一个特征的测试，树的分支代表该特征的每一个测试结果，而树的每一个叶子节点代表一个类别。树的最高层是就是根节点2022-01-01
%Y会被无世纪的年份所替代，%m会被01到12之间的一个十进制月份数替代，其他依次类推，%在python的格式化输出，有转换字符的作用，本文给大家讲解python中的% 是什么意思,起到什么作用，感兴趣的朋友一起看看吧2022-12-12
大家好，本篇文章主要讲的是Python装饰器中@property使用详解，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下2022-01-01
今天小编就为大家分享一篇Python+OpenCV实现将图像转换为二进制格式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
平常我都是直接执行 pip install 安装的第三方库，很多教程也是这么介绍的，一直以来我都认为这是标准的、正确的安装 Python 第三方库的姿势。下面小编给大家分享一篇教程，一起看看吧2019-11-11
这篇文章主要介绍了python内存管理机制原理详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
详解Django-channels 实现WebSocket实例
这篇文章主要介绍了详解Django-channels实现WebSocket实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-08-08
网络爬虫是一种按照一定的规则自动浏览、检索网页信息的程序或者脚本。网络爬虫能够自动请求网页，并将所需要的数据抓取下来。本文将为大家介绍如何利用爬虫获取某宝男装信息，感兴趣的小伙伴可以了解一下2021-12-12
这篇文章主要介绍了PyTorch中Tensor的拼接与拆分的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-08-08
最新评论