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

cuDNN unavailable #74

Closed
yhalk opened this issue Jul 15, 2015 · 1 comment
Closed

cuDNN unavailable #74

yhalk opened this issue Jul 15, 2015 · 1 comment

Comments

@yhalk
Copy link

yhalk commented Jul 15, 2015

Hello,

I want to use Mocha with a GPU backend, however the CUDA version that is installed on the system is 5.0 which does not support cuDNN.
I have seen this issue #52 and the only problem with my network is that it uses sigmoid neurons and a softmax layer, which need cuDNN.Is there a workaround to make it work even if computation with sigmoids and softmax becomes less efficient, eg make it use a CPU backend for those?

Thanks in advance for the help!

@pluskid
Copy link
Owner

pluskid commented Jul 15, 2015

@yhalk Unfortunately, currently we do not support mixed backend computing because data transfer between CPU and GPU is actually quite costly. And currently the GPU backend requires the cuDNN library even if cuDNN convolution is not used in the network. Maybe you could try to use CPU backend with native extension, which might be a little bit faster than the default julia backend depending on your network.

@pluskid pluskid closed this as completed Sep 21, 2015
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

2 participants