Я пытаюсь запустить нейронную сеть с pycaffe на gpu.Ошибка: ошибка == cudaSuccess (2 против 0) из памяти
Это работает, когда я вызываю сценарий в первый раз. Когда я запускаю тот же скрипт во второй раз, CUDA выдает ошибку в заголовке.
Размер партии 1, размер изображения на данный момент равен 243x322, gpu имеет 8gb RAM.
Я думаю, что мне не хватает команды, которая сбрасывает память?
спасибо!
EDIT:
Может быть, я должен прояснить несколько вещей: я бегу Caffe на окнах.
Когда я вызываю скрипт с python script.py, процесс завершается, и память gpu освобождается, поэтому это работает.
С IPython, который я использую для отладки, память GPU действительно не получает освобожден (после одного прохода, 6 из 8 Б.Г. в использовании, спасибо за NVidia-сми предложения!)
Итак, то, что я ищу это команда, которую я могу вызывать из pyhton, вдоль линий:
запуска сети
выходной образ процесса
бесплатно память GPU
Есть ли у вас другие процессы с использованием графического процессора? – Shai
, если он работает один раз и позже, вы получаете такую ошибку, которая, безусловно, не освобождается. Если это действительно превышает память, вы получите ее всегда. Также память может быть превышена из-за размера партии, размера изображения, размера ядра, размера выходного параметра, глубины сетевых уровней и т. Д. –
Как насчет использования команды 'del your_net' для освобождения сетевого объекта в python? – Dale