Как очистить память графического процессора, используемую моделью doctr ⇐ Python

[Anonymous]

Я использую doctr для целей распознавания... и модель doctr работает в графическом процессоре, поэтому после того, как работа по распознаванию завершена, она все равно загружается в память графического процессора.
Я пытаюсь очистить память графического процессора, используя

Код: Выделить всё

with torch.no_grad(): torch.cuda.empty_cache()

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

Подробнее здесь: https://stackoverflow.com/questions/787 ... octr-model