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 。