Как очистить память графического процессора Tensorflow-Keras? - PullRequest
0 голосов
/ 02 апреля 2019

Я подгоняю модель к циклу for, но получаю сообщение об ошибке, что память моего GPU заполнена. Я использую Keras в IDE Anaconda Spyder. Мой графический процессор - Asus GTX 1060 6 ГБ.

Я также использовал такие коды, как: K.clear_session(), gc.collect(), tf.reset_default_graph(), del custom_model, но ни один из них не работал. Свойства графического процессора говорят, что 98% памяти заполнено:

Ничего не очищает память GPU, кроме numba.cuda.close(), но не позволяет мне снова использовать мой графический процессор. Единственный способ очистить его - перезапустить ядро ​​и перезапустить мой код.

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

1 Ответ

1 голос
/ 20 апреля 2019

Я нашел решение в этой статье:

https://github.com/keras-team/keras/issues/12625

Спасибо всем, кто участвовал.

...