caffe内核死于jupyter qtconsole

Question

我在jupyter笔记本上运行http://nbviewer.jupyter.org/github/BVLC/caffe/blob/master/examples/01-learning-lenet.ipynb。

但是，在我运行代码之后，它弹出错误：kernel died

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

如何解决这个问题？

Answer 1

最后，我得到了答案。它是在文件lenet_auto_solver.prototxt的末尾添加一行代码solver_mode = GPU。