I'm trying to execute the caffe example script from
here Everything seems to be fine as I've installed caffe just few minutes back ..... But this message pops up when I run line 7
"The kernel appears to have died. It will restart automatically."
This is the code:
caffe.set_device(0)
caffe.set_mode_gpu()
### load the solver and create train and test nets
solver = None # ignore this workaround for lmdb data (can't instantiate two solvers on the same data)
solver = caffe.SGDSolver('mnist/lenet_auto_solver.prototxt')
I'm running and have compiled caffe on CPU_ONLY=1 .... so
I feel the set_mode_gpu() is creating a problem/...
What modifications should I make?