Как очистить память графического процессора после обучения модели PyTorch без перезапуска ядра ⇐ Python

[Anonymous]

Я обучаю модели глубокого обучения PyTorch на ноутбуке Jupyter-Lab, используя для обучения CUDA на графическом процессоре Tesla K80. При выполнении обучающих итераций используются 12 ГБ памяти графического процессора. Я заканчиваю обучение сохранением контрольной точки модели, но хочу продолжить использовать блокнот для дальнейшего анализа (анализа промежуточных результатов и т.п.).

Однако эти 12 ГБ продолжают использоваться занято (как видно из nvtop) после окончания обучения. Я хотел бы освободить эту память, чтобы использовать ее для других ноутбуков.

На данный момент мое решение — перезапустить ядро ​​этого ноутбука, но это не решает мою проблему. проблема, потому что я не могу продолжать использовать тот же блокнот и соответствующий результат, вычисленный до сих пор.

Подробнее здесь: https://stackoverflow.com/questions/578 ... ing-kernel