PyTorch 删除当前进程中分配的显存

在 PyTorch 中，可以使用以下代码来删除当前进程中分配的显存：

import torch

torch.cuda.empty_cache()

这将清除当前进程中未使用的缓存，并释放显存。这对于在训练模型的过程中释放显存非常有用。但是请注意，这不会释放已经被模型占用的显存，因此如果你想完全释放所有显存，你需要停止当前进程并重启 Python 环境。