Python 3.x Pytorch RNNLearner预测docker内存泄漏原因

Python 3.x Pytorch RNNLearner预测docker内存泄漏原因,python-3.x,docker,pytorch,tensor,fast-ai,Python 3.x,Docker,Pytorch,Tensor,Fast Ai,我试图用预先训练好的模型来运行一个简单的预测,但它一直在消耗记忆 [环境] docker:docker Ubuntu18.04,python3.6容器 Pytork: 法斯泰:1.0.46 这是python代码: from fastai.basic_train import load_learner predictor = load_learner(path, "file.pkl") pred_class, pred_idx, outputs = predictor.predict("stri

我试图用预先训练好的模型来运行一个简单的预测,但它一直在消耗记忆

[环境]

  • docker:docker Ubuntu18.04,python3.6容器
  • Pytork:
  • 法斯泰:1.0.46
  • 这是python代码:

    from fastai.basic_train import load_learner
    predictor = load_learner(path, "file.pkl")
    pred_class, pred_idx, outputs = predictor.predict("string")
    
    我无法在我的mac或linux机器上重现内存泄漏,因此我认为这是与docker相关的问题,可能是docker中缺少库或其他什么。有什么想法吗