pixtext 跑了100张图片后出现 OOM

[liuwei110]

只启动了一个进程跑，本来gpu只占了1.8G，跑完100多张图片后，直接打满预发环境申请的3g显存，然后后面就直接oom了，是否需要做任何释放动作？

[liuwei110]

后面换了批图片申请了10g的显存还是一直涨到10g

[sqbghb]

我现在也是这样，你解决了吗？

[breezedeus]

嗯。我找找原因

[breezedeus]

试试在每次调用完执行下下面的函数：

import torch
import gc

def clear_memory():
    """清理内存和显存"""
    if torch.cuda.is_available():
        torch.cuda.empty_cache()
    gc.collect()