yahoo/open_nsfw

GPU version not faster than CPU version

HolyB opened this issue 8 years ago · 3 comments

HolyB commented 8 years ago

Hi,
I tried GPU version of the open_nsfw. But I found the GPU version is just a little faster than CPU.
Does anyone know the reason?

jeshjesh commented 7 years ago

How much time did it take in classify?

jingith commented 7 years ago

Me too~~ GPU not faster than cpu， I watch nvidia-smi GPU-Util is still 0%，and GPU Memory is not zero. It means processing in GPU，but not use GPU‘calculate

lleiiell commented 5 years ago

try

caffe.set_mode_gpu()
caffe.set_device(0)

it works for me, up to 10 times faster