mini-caffe icon indicating copy to clipboard operation
mini-caffe copied to clipboard

Question with using GPU

Open ligongzheng opened this issue 8 years ago • 3 comments
trafficstars

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).

ligongzheng avatar Sep 19 '17 03:09 ligongzheng

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.

luoyetx avatar Sep 20 '17 04:09 luoyetx

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

ligongzheng avatar Sep 28 '17 08:09 ligongzheng

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

luoyetx avatar Sep 28 '17 08:09 luoyetx