仅在CPU环境下替换Torch.cuda.empty_cache()

时间:2019-11-12 14:54:15

标签: python pytorch

当前,我正在使用仅支持CPU的PyTorch。当我运行推理时,该输入文件的信息会以某种方式存储在缓存中,并且用于推理的每个新的唯一文件的内存都会不断增加。另一方面,如果我一次又一次使用同一文件,内存使用不会增加。

只有在使用CPU的情况下,才有清除cuda.empty_cache()之类的缓存的方法。

0 个答案:

没有答案