mxnet-ssd.cpp icon indicating copy to clipboard operation
mxnet-ssd.cpp copied to clipboard

400ms on gpu/1060 which is much slow than yolo

Open anguoyang opened this issue 8 years ago • 11 comments

i have tested on your demo on gpu mode, it took about 400ms for 1 image, which is much slow than yolo(<100ms).

anguoyang avatar Apr 09 '17 11:04 anguoyang

  1. The default config is not using cuda and cudnn. you should check your config.
  2. As in my case,the first time of running will take more time than usual. you can test more.

nopattern avatar Apr 10 '17 04:04 nopattern

hi,thank you, I have checked the pre-defined values before the main, and debug to see the device type and device id, they are OK. and i also added clock related lines before and after the detect function to calculate the duration only on the detection. is there any other configurations?

anguoyang avatar Apr 10 '17 09:04 anguoyang

you can just use normal mxnet-ssd if you want to run on gtx1060, Why to use this version? This version seems that it supports cuda but doesn't support cudnn .

nopattern avatar Apr 10 '17 13:04 nopattern

because I want to use mxnet in c++

anguoyang avatar Apr 10 '17 20:04 anguoyang

How did you get GPU to work using this repo? I think the makefile don't support GPU yet.

zhreshold avatar Apr 10 '17 22:04 zhreshold

I modify on visual studio to run it, GPU is working,however, cudnn is not working

anguoyang avatar Apr 10 '17 22:04 anguoyang

Can you profile the runtime?

zhreshold avatar Apr 10 '17 22:04 zhreshold

profile?

anguoyang avatar Apr 10 '17 23:04 anguoyang

The default config about cuda is #define MSHADOW_USE_CUDA 0, As @threshold ,the default seems don't support cuda. You can use c_predict_api.h to use gtx1060 in c++ .

nopattern avatar Apr 11 '17 03:04 nopattern

@anguoyang how did you modify it on visual studio to work with GPU? I'd gladly accept any pointers to be able to do the same

pelre avatar Apr 13 '17 20:04 pelre

@zhreshold i can not get gpu to work ,could you give me some tips to get gpu to work? thanks

aa12356jm avatar Dec 20 '17 08:12 aa12356jm