python-3.x - Pytorch RNNLearner predict 导致 docker 内存泄漏

Question

我正在尝试使用预训练模型运行一个简单的预测，但它一直在消耗内存。

[环境]

1. docker：docker Ubuntu18.04，python3.6容器
2. pytorch： https ://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
3. 快泰：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 或其他一些缺少的库。任何想法？