Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

how to enable GPU in the build #2

Open
kaishijeng opened this issue Oct 19, 2016 · 5 comments
Open

how to enable GPU in the build #2

kaishijeng opened this issue Oct 19, 2016 · 5 comments

Comments

@kaishijeng
Copy link

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

Thanks,

@zhreshold
Copy link
Owner

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.

@zhyj3038
Copy link

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

@zhreshold
Copy link
Owner

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

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

@yang-fei
Copy link

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.

@aa12356jm
Copy link

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

5 participants