当前位置: 首页 > news >正文

Pytorch:torch.cuda.empty_cache()

torch.cuda.empty_cache()

原理

torch.cuda.empty_cache() 是PyTorch中用来释放未被分配的缓存的内存的函数在使用GPU进行计算时,CUDA会在内部维护一个内存缓存池,以便更快地分配和释放。但有时候,这些缓存的内存在不再需要的时候,它们不会被自动返回到操作系统。

torch.cuda.empty_cache() 会释放这些未被使用的缓存内存,所以其他程序或者同一程序的其他部分可以使用这些资源。

用法

# 当感觉到GPU内存紧张或者在删除了大量数据之后,可以调用
import torch
torch.cuda.empty_cache()

相关文章:

  • uniapp 开发微信小程序 出现启用组件按需注入问题如何解决
  • 基于Springboot的防疫物资管理信息系统(有报告)。Javaee项目,springboot项目。
  • 【论文阅读】Masked Autoencoders Are Scalable Vision Learners
  • Docker常用命令!!!
  • 九.pandas绘图基础
  • Python基础----数据容器(持续更新中)
  • QGIS编译(跨平台编译)056:PDAL编译(Windows、Linux、MacOS环境下编译)
  • Learn OpenGL 19 几何着色器
  • java 通过 IMetaStoreClient 取 hive 元数据信息
  • 基于ArcGIS等多技术融合提升环境、生态、水文、土地、农业、大气等领域科研技术应用
  • Spring底层核心原理解析
  • 【晴问算法】提高篇—动态规划专题—01背包问题
  • Android: Gradle 命令
  • Form当中method的post和get的区别?
  • 使用光标精灵更换电脑鼠标光标样式,一键安装使用
  • #Java异常处理
  • JavaScript服务器推送技术之 WebSocket
  • JS数组方法汇总
  • Vue学习第二天
  • Wamp集成环境 添加PHP的新版本
  • 前端之Sass/Scss实战笔记
  • 手写双向链表LinkedList的几个常用功能
  • 微服务框架lagom
  • 用 Swift 编写面向协议的视图
  • 用jquery写贪吃蛇
  • - 转 Ext2.0 form使用实例
  • 关于Kubernetes Dashboard漏洞CVE-2018-18264的修复公告
  • 扩展资源服务器解决oauth2 性能瓶颈
  • #Ubuntu(修改root信息)
  • #我与Java虚拟机的故事#连载07:我放弃了对JVM的进一步学习
  • $ is not function   和JQUERY 命名 冲突的解说 Jquer问题 (
  • %check_box% in rails :coditions={:has_many , :through}
  • (ZT)一个美国文科博士的YardLife
  • (板子)A* astar算法,AcWing第k短路+八数码 带注释
  • (黑客游戏)HackTheGame1.21 过关攻略
  • (四)搭建容器云管理平台笔记—安装ETCD(不使用证书)
  • (一) springboot详细介绍
  • (转)机器学习的数学基础(1)--Dirichlet分布
  • (转)拼包函数及网络封包的异常处理(含代码)
  • (最简单,详细,直接上手)uniapp/vue中英文多语言切换
  • .NET CLR Hosting 简介
  • .net Signalr 使用笔记
  • .NET 应用架构指导 V2 学习笔记(一) 软件架构的关键原则
  • .Net 中Partitioner static与dynamic的性能对比
  • .NET精简框架的“无法找到资源程序集”异常释疑
  • /3GB和/USERVA开关
  • @requestBody写与不写的情况
  • [codevs 1296] 营业额统计
  • [CUDA 学习笔记] CUDA kernel 的 grid_size 和 block_size 选择
  • [Go WebSocket] 多房间的聊天室(五)用多个小锁代替大锁,提高效率
  • [HNOI2010]BUS 公交线路
  • [leetcode] 103. 二叉树的锯齿形层次遍历
  • [leetcode] Multiply Strings
  • [LeetCode]—Simplify Path 简化路径表达式
  • [NHibernate]条件查询Criteria Query