caffe kernel死了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

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