pytorch清除cuda缓存

在 PyTorch 中，可以使用 `torch.cuda.empty_cache()` 方法来清除 CUDA 缓存。这个方法会释放 PyTorch 占用的所有显存，但并不会影响已经存储在显存中的变量或张量。

使用该方法的示例代码如下：

import torch

# 创建一些张量并进行计算
x = torch.randn(1000, 1000).cuda()
y = torch.randn(1000, 1000).cuda()
z = torch.matmul(x, y)

# 清除 CUDA 缓存
torch.cuda.empty_cache()

在执行大量的计算、训练模型等占用显存较多的任务时，建议定期清理 CUDA 缓存，以避免显存不足的问题。