-
Notifications
You must be signed in to change notification settings - Fork 18.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
CPU only mode in Caffe #2241
Comments
I get the same error when reproducing the steps from https://github.com/BVLC/caffe/tree/master/examples/finetune_flickr_style in CPU-mode. "Now we can train! (You can fine-tune in CPU mode by leaving out the -gpu flag.)" caffe % ./build/tools/caffe train -solver models/finetune_flickr_style/solver.prototxt -weights models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel I0403 18:09:40.822396 7323 caffe.cpp:113] Use GPU with device ID 0 |
models/finetune_flickr_style/solver.prototxt, uncomment the last line. On Fri, Apr 3, 2015 at 9:42 PM, Simon Brugman notifications@github.com
|
The problem is solved, in CPU mode we should not call |
@azamosavi |
@hana9090 |
If you choose "only cpu" mode, except you should delete "#" in "# CPU_ONLY := 1" of Makefile.config. and you also should modify something in your example. For example, if you train MNIST, you should modify file "lenet_solver.prototxt" in /your caffe path/examples/MNIST, in the last line of this file ,you should modify solver mode to be CPU. |
...and how do you "modify solver to be CPU"? Thank you |
OK, just found it. Never mind....:P |
i have the same problem can someone help me plz |
We have installed Caffe and all its prerequisites except CUDA on Fedora. We want to run Caffe in CPU mode and we have made changes to support this by changing the config file during installation as well as prototxt files.
However, when we run the Matlab script calling Caffe:
caffe('set_device', 1);
we get the following error:
Logging before InitGoogleLogging() is written to STDERR
F0401 22:36:01.857090 18173 common.cpp:55] Cannot use GPU in CPU-only Caffe: check mode.
I was wondering if anyone can help us CPU only mode for Caffe in Matlab.
The text was updated successfully, but these errors were encountered: