This repository has been archived by the owner on Oct 18, 2019. It is now read-only.

GPU version not faster than CPU version #33

Open

HolyB opened this issue Dec 8, 2016 · 3 comments

HolyB commented Dec 8, 2016

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 May 9, 2017

How much time did it take in classify?

jingith commented Jun 29, 2017

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 Sep 18, 2019 •

edited

Loading

try

caffe.set_mode_gpu()
caffe.set_device(0)

it works for me, up to 10 times faster

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.