ubuntu下训练模型,内存不断累积且不释放,直到占满,进程被杀死

如果您使用 Jittor 时遇到了安装问题、运行错误、输出错误等问题,您可以在 “问答” 区提问。
为了更好地定位您的问题,建议包含以下内容:

  1. 错误信息、
    ubuntu下训练模型,内存不断累积且不释放,直到占满,进程被杀死。在pytorch下训练时,内存占约8g在jittor上训练时,内存随着训练次数上涨,由8g逐步占满内存。进程在内存被占满后卡顿阻塞,显示“已杀死”。训练代码中也不存在全局变量,loss使用时也使用了item。
  2. Jittor 运行的完整 log(建议复制粘贴运行输出,如果可能请不要使用截图)
    Epoch74/2481
    1.8612831830978394
    Epoch74/2482
    1.696168303489685
    Epoch74/2483
    1.8896644115447998
    Epoch74/2484
    2.0000035762786865
    Epoch74/2485
    1.7134026288986206
    Epoch74/2486
    2.098353385925293
    Epoch74/2487
    1.7821273803710938
    Epoch74/2488
    1.8932944536209106
    Epoch74/2489
    1.9438241720199585
    Epoch74/2490
    已杀死
  3. 复现此问题的代码或者描述、
  1. 您认为正确结果应当如何、
  2. 其他必要信息

这有一个 参考模板