标签: python pytorch
当前,我正在使用仅支持CPU的PyTorch。当我运行推理时,该输入文件的信息会以某种方式存储在缓存中,并且用于推理的每个新的唯一文件的内存都会不断增加。另一方面,如果我一次又一次使用同一文件,内存使用不会增加。
只有在使用CPU的情况下,才有清除cuda.empty_cache()之类的缓存的方法。