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 。