search

alexgkendall / SegNet-Tutorial

Files for a tutorial to train SegNet for road scenes using the CamVid dataset
http://mi.eng.cam.ac.uk/projects/segnet/tutorial.html
847 stars 517 forks source link

Cannot use GPU in CPU-only Caffe: check mode. #59

Closed WeiyiLi closed 7 years ago

WeiyiLi commented 7 years ago

When I tried to run Live Demo, I got the error:

F0822 21:57:53.386196 38133 conv_layer.cpp:71] Cannot use GPU in CPU-only Caffe: check mode. *** Check failure stack trace: *** Aborted (core dumped)

Can anyone help me the fix this problem? Where can I change the mode? Thank you very much!

WeiyiLi commented 7 years ago

Fixed! In line 29 of webcam_demo.py, change caffe.set_mode_gpu() to caffe.set_mode_cpu()