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

Question with using GPU #51

Open
ligongzheng opened this issue Sep 19, 2017 · 3 comments
Open

Question with using GPU #51

ligongzheng opened this issue Sep 19, 2017 · 3 comments

Comments

@ligongzheng
Copy link

Hello,
Why the GPU is used when I didn't invoke caffe::SetMode(caffe::GPU,0) in my program ? Although I complied it with options -DUSE_CUDA=ON -DUSE_CUDNN=ON, the value of mode_ is Caffe::CPU and it should work in CPU style (I get it from common.cpp). In my thought,even though I complied it with cuda and cudnn on, the mode is still Caffe::CPU if I didn't invoke the SetMode function and GPU should not be occupied (I check it with NVIDIA-SMI).

@luoyetx
Copy link
Owner

luoyetx commented Sep 20, 2017

Yes, It's a bug. Checkout branch fix-create-layer. I haven't merge it yet. The bug is from cudnn layers, when you build with -DUSE_CUDNN=ON.

@ligongzheng
Copy link
Author

Thank you for your answer . it seems like the page is not found!

@luoyetx
Copy link
Owner

luoyetx commented Sep 28, 2017

Sorry, I merged the branch. The code is now in master branch.

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

2 participants