caffe kernel умер jupyter qtconsole

Question

Запуск http://nbviewer.jupyter.org/github/BVLC/caffe/blob/master/examples/01-learning-lenet.ipynb на ноутбуке jupyter.

Однако, выскакивает ошибка: kernel died, после того, как я запускаю код

solver = caffe.SGDSolver('mnist/lenet_auto_solver.prototxt') 

Как решить эту проблему?

Answer 1

Наконец-то я получил ответ. Он должен добавить строку кода solver_mode = GPU в конце файла lenet_auto_solver.prototxt.