Pytorch：torch.cuda.empty_cache()

北方骑马的萝卜

于 2024-03-20 10:07:30 发布

阅读量3.9k

点赞数 2

分类专栏： Pytorch系列文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/weixin_42046845/article/details/136867683

版权

26 篇文章

订阅专栏

torch.cuda.empty_cache()

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

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

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