caffe错误cudnn版本问题

﹏ヽ暗。殇╰゛Y 2022-05-28 06:10 215阅读 0赞

在编译caffe时出现错误：

**\[plain\]** [view plain][] [copy][view plain]

1.  In file included from ./include/caffe/util/device\_alternate.hpp:40:0, 
2.   from ./include/caffe/common.hpp:19, 
3.   from src/caffe/common.cpp:7: 
4.  ./include/caffe/util/cudnn.hpp: In function ‘void caffe::cudnn::createPoolingDesc(cudnnPoolingStruct\*\*, caffe::PoolingParameter\_PoolMethod, cudnnPoolingMode\_t\*, int, int, int, int, int, int)’: 
5.  ./include/caffe/util/cudnn.hpp:127:41: error: too few arguments to function ‘cudnnStatus\_t cudnnSetPooling2dDescriptor(cudnnPoolingDescriptor\_t, cudnnPoolingMode\_t, cudnnNanPropagation\_t, int, int, int, int, int, int)’ 
6.   pad\_h, pad\_w, stride\_h, stride\_w)); 
7.   ^ 
8.  ./include/caffe/util/cudnn.hpp:15:28: note: in definition of macro ‘CUDNN\_CHECK’ 
9.   cudnnStatus\_t status = condition; \\ 
10.  ^ 
11. In file included from ./include/caffe/util/cudnn.hpp:5:0, 
12.  from ./include/caffe/util/device\_alternate.hpp:40, 
13.  from ./include/caffe/common.hpp:19, 
14.  from src/caffe/common.cpp:7: 
15. /usr/local/cuda-7.5//include/cudnn.h:803:27: note: declared here 
16.  cudnnStatus\_t CUDNNWINAPI cudnnSetPooling2dDescriptor( 
17.  ^ 
18. make: \*\*\* \[.build\_release/src/caffe/common.o\] Error 1

![Center][]

![Center 1][]

这是因为当前版本的caffe的cudnn实现与系统所安装的cudnn的版本不一致引起的。

解决办法：

1.将./include/caffe/util/cudnn.hpp 换成最新版的caffe里的cudnn的实现，即相应的cudnn.hpp.

2. 将./include/caffe/layers里的，所有以cudnn开头的文件，例如cudnn\_conv\_layer.hpp。 都替换成最新版的caffe里的相应的同名文件。

3.将./src/caffe/layer里的，所有以cudnn开头的文件，例如cudnn\_lrn\_layer.cu，cudnn\_pooling\_layer.cpp，cudnn\_sigmoid\_layer.cu。

都替换成最新版的caffe里的相应的同名文件。

[view plain]: https://blog.csdn.net/u011070171/article/details/52292680#
[Center]: /images/20220528/ee4de838016f4a9392429fba1a05c754.png
[Center 1]: /images/20220528/6abb3cfd36884527855e5dd3c1051fa2.png