8000 add support for cuDNN v7 by icyblade · Pull Request #5972 · BVLC/caffe · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

add support for cuDNN v7 #5972

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

Merged
merged 1 commit into from
Nov 7, 2017
Merged

add support for cuDNN v7 #5972

merged 1 commit into from
Nov 7, 2017

Conversation

icyblade
Copy link
Contributor

This PR will try to add support for cuDNN v7.
Tested on Tesla K80/Ubuntu 16.04/CUDA 9.0/cuDNN v7.

@wjcskqygj2015
Copy link

Perfectly, I originally try to pull this request

@wjcskqygj2015
Copy link

However, in order to fit cuda 9.0, we might need to anotate
line 64 in cuda/include/crt/common_functions.h since the preprocessor is not worked well
#define CUDACC_VER "CUDACC_VER is no longer supported. Use CUDACC_VER_MAJOR, CUDACC_VER_MINOR, and CUDACC_VER_BUILD instead."

@Noiredd
Copy link
Member
Noiredd commented Nov 7, 2017

From a few tests I ran myself (before and after an update to cuDNN 7), this works well. Thank you for the contribution, @icyblade!

@Noiredd Noiredd merged commit c430690 into BVLC:master Nov 7, 2017
XinYao1994 pushed a commit to XinYao1994/caffe that referenced this pull request Aug 29, 2018
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

Successfully merging this pull request may close these issues.

3 participants
0