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

Support cuDNN v4 #13

Open
andravin opened this issue Jan 12, 2016 · 9 comments
Open

Support cuDNN v4 #13

andravin opened this issue Jan 12, 2016 · 9 comments

Comments

@andravin
Copy link
Collaborator

Placeholder for cuDNN v4 support development effort.

I am only set up to test on Linux, so it would be helpful if somebody else could test on Windows and Mac.

@kashif
Copy link
Collaborator

kashif commented Apr 1, 2016

@andravin happy to look at this issue, is there anything specific you want me to test? thanks!

@andravin
Copy link
Collaborator Author

andravin commented Apr 2, 2016

Hi @kashif I was implementing cuDNN v4 in this fork: https://github.com/andravin/cudnn-python-wrappers

Some things are probably not implemented, been a couple months since I looked at it.

@kashif
Copy link
Collaborator

kashif commented Apr 4, 2016

ok thanks @andravin I'll use that and get it working on mac and update it as well...

@andravin
Copy link
Collaborator Author

andravin commented Apr 4, 2016

The first thing to check for each platform is whether the library name specified here is correct:

if sys.platform in ('linux2', 'linux'):
    _libcudnn_libname_list = ['libcudnn.so', 'libcudnn.so.4', 'libcudnn.so.4.0.4']
elif sys.platform == 'darwin':
    # TODO: check darwin version
    _libcudnn_libname_list = ['libcudnn.dylib', 'libcudnn.7.5.dylib']
elif sys.platform == 'win32':
    # TODO: check win32 version
    _libcudnn_libname_list = ['cudnn70_64.dll']
else:
    raise RuntimeError('unsupported platform')

@kashif kashif mentioned this issue Apr 6, 2016
@hannes-brt
Copy link
Owner

Thanks for contributing @kashif and @andravin. I am currently not using this library myself and therefore also not actively developing it either. If either one of you would like to take stewardship of this project, I'd be more than happy to give you push access to the repo.

@andravin
Copy link
Collaborator Author

andravin commented Apr 7, 2016

I am not actively using it either, but I am happy to help out when I can.

I want to stress that I really only tested the forward path convolution functions with regards to cuDNN support, so really everything else needs a bit of testing. @kashif have you gotten more code coverage in your testing?

@kashif
Copy link
Collaborator

kashif commented Apr 7, 2016

@andravin I plan to add more tests and perhaps i will ask you about them when I start. @hannes-brt I would be happy if @andravin takes ownership since that way I can run things by him when I do stuff...

@hannes-brt
Copy link
Owner

I added both of you as collaborators on the repo.

@kashif
Copy link
Collaborator

kashif commented Apr 7, 2016

@hannes-brt thanks!

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

3 participants