-
Notifications
You must be signed in to change notification settings - Fork 745
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
added make options for gpu build to mxnet #609
Conversation
Thanks! That's a good start. Let's include it as an "extension" just like with Caffe and TensorFlow (check their |
I have added the extension handling but I am not sure how to test it. Please check if CUDA home needs to be changed |
Thanks! Let's wait and see what Travis CI has to say about it... |
The build passes, but it doesn't appear to be compiling any CUDA kernels...? |
I think the problem is the different CUDA_HOME path. I have seen that the cuda project expects the standard /usr/local/cuda, I had /opt/cuda so it worked only on arch. Sorry for missing that. |
Looks like the build is failing:
https://travis-ci.org/bytedeco/javacpp-presets/jobs/421178460 BTW, it's basically building inside a Docker container using "nvidia/cuda:9.2-cudnn7-devel-centos7" as image, so you could run with that locally to make sure it works alright. |
Using the release archives seems to have fixed that issue! |
I have added options for gpu / cuda build.
To use cuda set CUDA_HOME to the folder containing cuda (e.g. /opt/cuda for arch linux) in the environment of the build
To use cudnn set USE_CUDNN=1
The has only been tested on arch linux_x86_64 where it compiles