caffe 编译error: too few arguments to function ‘cudnnStatus_t cudnnSetPooling2dDescriptor

最新推荐文章于 2022-11-04 00:43:05 发布

原创最新推荐文章于 2022-11-04 00:43:05 发布 · 2k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

本文记录了在编译Caffe深度学习框架过程中遇到的一个具体错误，并提供了详细的错误信息及可能的解决方案。该错误出现在尝试编译特定层时，涉及cuDNN相关设置不当的问题。

出现：

In file included from ./include/caffe/util/device_alternate.hpp:40:0,
                 from ./include/caffe/common.hpp:19,
                 from ./include/caffe/blob.hpp:8,
                 from ./include/caffe/layers/infogain_loss_layer.hpp:6,
                 from src/caffe/layers/infogain_loss_layer.cpp:5:
./include/caffe/util/cudnn.hpp: In function ‘void caffe::cudnn::createPoolingDesc(cudnnPoolingStruct**, caffe::PoolingParameter_PoolMethod, cudnnPoolingMode_t*, int, int, int, int, int, int)’:
./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)’
         pad_h, pad_w, stride_h, stride_w));