Pytorch RNNLearner предсказать вызывает утечку памяти в докере - PullRequest
Новая
       19

Pytorch RNNLearner предсказать вызывает утечку памяти в докере

1 голос
/ 03 мая 2019

Я пытаюсь выполнить простой прогноз с предварительно обученной моделью, но он продолжает поглощать память.

[ENV]

  1. докер: докер Ubuntu18.04, контейнер python3.6
  2. pytorch: https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
  3. fastai: 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 или STH еще. Есть мысли?

...