咖啡仁死了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。