yahoo / open_nsfw

Not Suitable for Work (NSFW) classification using deep neural network Caffe models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GPU version not faster than CPU version

HolyB opened this issue · comments

commented

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?

commented

How much time did it take in classify?

commented

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

commented

try

caffe.set_mode_gpu()
caffe.set_device(0)

it works for me, up to 10 times faster