Cannot use GPU in CPU-only Caffe: check mode

bear63 / sceneReco

ctpn+crnn Scene character recognition

943 stars 371 forks source link

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

Closed ly2251 closed 6 years ago

ly2251 commented 6 years ago

I build caffe CPU Only

root@ubuntu:/home/user/sceneReco# python demo.py WARNING: Logging before InitGoogleLogging() is written to STDERR F0410 03:50:20.793697 5050 common.cpp:66] Cannot use GPU in CPU-only Caffe: check mode. Check failure stack trace: Aborted (core dumped

pengpengO commented 6 years ago

I change the ctpnport.py like this:

caffe.set_mode_cpu()

bear63 / sceneReco

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

caffe.set_device(cfg.TEST_GPU_ID)