Skip to content

Comments

add support for cuDNN v7#5972

Merged
Noiredd merged 1 commit intoBVLC:masterfrom
icyblade:icyblade-patch-1
Nov 7, 2017
Merged

add support for cuDNN v7#5972
Noiredd merged 1 commit intoBVLC:masterfrom
icyblade:icyblade-patch-1

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