gpu - 如何释放 GPU 上的空间？

Question

我在一个非常大的数据集（大小约为 1TB）上处理我的模型，方法是将其分成块，然后对每个块运行预测，然后将结果附加到文件中。

在每个块上运行后，我必须释放 GPU 的内存，以便可以处理和预测另一个块。

如果存在清除 GPU 内存的命令，我可以运行循环，而不是一次又一次地运行“重新启动内核”。