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()

相关推荐

最近更新

  1. TCP协议是安全的吗?

    2024-03-24 18:48:01       14 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-03-24 18:48:01       16 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-03-24 18:48:01       15 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-03-24 18:48:01       18 阅读

热门阅读

  1. trt | torch2trt的使用方式

    2024-03-24 18:48:01       19 阅读
  2. P7480 Reboot from Blue 线段树优化建图跑最短路

    2024-03-24 18:48:01       17 阅读
  3. 和neo4j相似的python工具

    2024-03-24 18:48:01       18 阅读
  4. 【2024-03-20】华为春招笔试三道编程题解

    2024-03-24 18:48:01       16 阅读
  5. C# System.Console.WriteLine的格式化输出

    2024-03-24 18:48:01       17 阅读
  6. C#面:简述可空类型

    2024-03-24 18:48:01       17 阅读
  7. 【2024-03-18】顺丰春招笔试两道编程题解

    2024-03-24 18:48:01       19 阅读
  8. 【串口开发】android 智能设备开发 知识笔记

    2024-03-24 18:48:01       21 阅读
  9. 学习笔记 | 微信小程序项目day06

    2024-03-24 18:48:01       22 阅读
  10. mysql基础02

    2024-03-24 18:48:01       15 阅读