caffe错误cudnn版本问题

在编译caffe时出现错误:

[plain]  view plain  copy
  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  



这是因为当前版本的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里的相应的同名文件。


发布了21 篇原创文章 · 获赞 15 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/liumingchun13/article/details/79761286