如何在GPU上释放空间?

时间:2018-06-08 12:59:44

标签: gpu gpgpu h2o h2o4gpu

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

在每个块上运行之后,我必须释放GPU的内存,以便可以处理和预测其他块。

而不是运行"重启内核"一次又一次,如果存在清除GPU内存的命令,我可以运行循环。

0 个答案:

没有答案