Dump GPU memory в CNTK Перевод художественного стиля

Я просматриваю Artistic Styling transfer tutorial in CNTK. Я могу пройти через одно изображение, затем перезапустить jupyter-ноутбук, чтобы избежать нехватки памяти GPU, чтобы создать дополнительное изображение. Код ошибки - удар. Я хотел бы просто поместить последние 2 блока кода в цикл for, который читает в папке с изображениями и создает новое стилизованное изображение для каждого изображения. Я думаю, что мне нужно сделать, это сбросить память GPU. Если это правильно, как мне это сделать?Dump GPU memory в CNTK Перевод художественного стиля

RuntimeError: CUDA failure 2: out of memory ; GPU=0 ; hostname=JOHN ;  expr=cudaMalloc((void**) &deviceBufferPtr, sizeof(AllocatedElemType) * numElements)

источник

2017-02-17 user18101

Перед обработкой второго изображения могли бы вы попробуйте позвонить del content_activations del loss import gc gc.collect() Это объекты, которые должны будут быть пересмотрены в любом случае, и таким образом, вы очистить их (поэтому освобождая память GPU) до следующего content_activations и loss есть были построены.

источник

2017-02-17 23:48:46

Я использовал этот метод, и он сделал это для первых двух изображений, но он ошибся на третьем. Я не уверен, что это лучше работает из-за удаления этих переменных, или я использую свой рабочий стол с 12 ГБ vram vs 4gb. В любом случае это не демпинг vram. Я хотел бы иметь возможность создать около 500 выводов. – user18101

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

Это исправлено у последнего мастера. Следующий релиз (2.0 beta 12), запланированный на 22 февраля 2017 года, будет иметь эти изменения. Я смог запустить 100 итераций, не выходя за рамки 1,5 ГБ vram. В моем тестовом случае у меня есть инструкции del и вызов сборщика мусора, как в (обновленном) ответе. Я не уверен, что они нужны. –

-1

Вы можете использовать другие рамки, кроме CNTK? Вы можете просто обучать свои модели переводом стиля в качестве сервисного сайта, например, http://somatic.io

источник

2017-02-20 00:17:44 jas

Неприемлемый ответ. –

Dump GPU memory в CNTK Перевод художественного стиля

ответ

Смежные вопросы