CuDNN 7 cases added, warnings removed#5869
CuDNN 7 cases added, warnings removed#5869gineshidalgo99 wants to merge 1 commit intoBVLC:masterfrom
Conversation
|
Travis builds worked nicely. I only added the case for cuDNN > 7. And I used |
|
Side note: I've been using it for the last 2 weeks (and I have 2 servers, one with cuDNN 7 and one with 5.1), and the same code works nicely in both versions. |
|
@shelhamer We've got 3 PRs for cuDNN 7 support: this one (which was first), #5879 (which also adds grouped convolution) and #5972 (which also updates installation instructions and Travis dependencies). Personally, I think the combination of the latter two would be best - what do you think? |
|
I decided to merge #5972 instead, as it also changes Travis to test with cuDNN 7. Offering any support for the new version is too important to keep us waiting for review of #5879. Still, thank you for your contribution @gineshidalgo99! |
|
when using cudnn 7 , the variables in 'cudnn.h' are not defined in all of cudnn layers. Is there anyone see this error? |
Added some cuDNN 7 support.
Now it works without any warnings in both cuDNN 7 and cuDNN < 7.
Otherwise, it pop ups the following warnings in cuDNN 7 (I'm enabling all warnings):
The code difference is adding these cases:
Highly tested and it's being currently used in OpenPose