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

how to enable GPU in the build

Open kaishijeng opened this issue 9 years ago • 5 comments

I don't see a flag to enable GPU in the install.sh. Do you know how to enable it?

Thanks,

kaishijeng avatar Oct 19 '16 21:10 kaishijeng

This repo is intended for minimal dependency, enabling GPU is somehow against this goal. If you would like to use GPU, use python version instead please. If mxnet release building with cuda in amalgamation in the future, I will re-consider this problem again.

zhreshold avatar Oct 19 '16 22:10 zhreshold

how fast mxnet-ssd.cpp detection object in image(eg.VOC2007) without GPU? thanks for your work very much!

zhyj3038 avatar Dec 13 '16 09:12 zhyj3038

Depending on the model/cpu/compilation mode, the run time significantly varies.

VGG16 model + 2.5GHz single thread with OPENMP disabled = 6s / image

zhreshold avatar Dec 13 '16 17:12 zhreshold

I just wonder have you built this with GPU enabled? The demo instruction has the flag --gpu=INT. But when I run this, I got a error "D:\Dev\Projects\mxnet\dmlc-core\include\dmlc/logging.h:235: [19:14:50] D:\Dev\Projects\mxnet\src\storage\storage.cc:50: Please compile with CUDA enabled"

But I didn't see any place to enable compiling with CUDA enabled.

I have installed CUDA8.0 with cudnn v5.1 on Win10. NVIDIA Samples run well.

yang-fei avatar Oct 30 '17 11:10 yang-fei

@feiyang-nottingham the author alreally said: the repo don't support gpu

aa12356jm avatar Dec 19 '17 06:12 aa12356jm