Skip to content
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

Predict by CPU #19

Open
xurong1981 opened this issue Nov 15, 2017 · 6 comments
Open

Predict by CPU #19

xurong1981 opened this issue Nov 15, 2017 · 6 comments

Comments

@xurong1981
Copy link

I want to use CPU-only to make a edge detection by rcf, and I have revised Makefile.config to commet "USE_CUDNN := 1" and unconment "CPU_ONLY := 1", and re-build and re-install caffe. But when I run predict, one error occured like the followings:


F1115 15:03:29.547533 6127 cudnn_relu_layer.cpp:13] Check failed: status == CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INITIALIZED

Is there anyone know how to solve such a problem ?
Besides, the model I fine-tuned is trained by GPU.

@yun-liu
Copy link
Owner

yun-liu commented Nov 21, 2017

The default engine in prototxt is CUDNN. You can solve this problem by removing the engine setting: (engine: CUDNN).

@xurong1981
Copy link
Author

Thank you for your reply. Actually, I changed "engine: CUDNN" by "engine: CAFFE", but it could not work. Is it available to remove ?

@yun-liu
Copy link
Owner

yun-liu commented Nov 21, 2017

Yes, you can remove it. The default setting in caffe will decide whether to use CPU or GPU automatically.

@xurong1981
Copy link
Author

I see, I will make a try. Thank you.

@yun-liu
Copy link
Owner

yun-liu commented Nov 21, 2017

You are welcome!

@edWin-m
Copy link

edWin-m commented Feb 28, 2018

Make sure you have installed cudnn correctly. i was having the same error before i fixed my cuDNN. Check the link below for the instructions.
http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installlinux-deb

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants