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 PyQt5中拖放效果的实现原理与实现代码，文中的示例代码讲解详细，感兴趣的小伙伴可以了解一下2022-11-11
这篇文章主要介绍了CentOS下Python3的安装及创建虚拟环境的方法,本文给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下2018-11-11
这篇文章主要为大家详细介绍了python实现简易五子棋游戏，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-05-05
这篇文章主要介绍了Python发送email的3种方法,本文讲解了使用登录邮件服务器方法、调用sendmail命令、使用smtp服务来发送三种方法,需要的朋友可以参考下2015-04-04
简单介绍Python中的try和finally和with方法
这篇文章主要介绍了Python中的try和finally和with方法,是Python学习当中的基础知识,需要的朋友可以参考下2015-05-05
本文主要介绍了详解pytest+Allure搭建方法以及生成报告常用操作，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-09-09
这篇文章主要为大家详细介绍了如何利用Python语言实现超快窗口截图功能，可以自动获取当前活动窗口并展示截图，感兴趣的可以了解一下2022-05-05
近来在学习用Python进行一些电脑运维的工作。所以下面这篇文章主要给大家介绍了关于利用python为运维人员写一个监控脚本的相关资料，文中通过示例代码介绍的非常详细，需要的朋友可以参考借鉴，下面随着小编来一起学习学习吧。2018-03-03
在Anaconda中如果没有指定路径,虚拟环境会默认安装在anaconda所安装的目录下,这篇文章主要给大家介绍了关于anaconda虚拟环境默认路径更改的相关资料,需要的朋友可以参考下2023-10-10
这篇文章主要介绍了Python爬虫爬取博客实现可视化,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
最新评论