CUDA 内存不足 - 是否需要重新启动？

Question

在 NVIDIA GPU 上使用 Pytorch（使用 Pycharm）编程时，我经常收到错误“CUDA 内存不足”，特别是当我手动中断进程 (KeyboardInterrupt) 时。然后我经常需要重启电脑来释放资源。

nvidia-smi 不显示占用空间的进程，并且报告的内存使用情况与 CUDA 错误期间报告的内存使用情况不同。

有没有什么办法可以在不需要重启的情况下释放内存？