caffe 编译错误：too few argument in function call

最新推荐文章于 2024-02-04 15:48:24 发布

haha074

最新推荐文章于 2024-02-04 15:48:24 发布

阅读量2.2k

点赞数 1

分类专栏： caffe 文章标签： cudnn caffe

本文链接：https://blog.csdn.net/haha074/article/details/89848695

版权

caffe 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

error log：

caffe/include/caffe/util/cudnn.hpp(112): error: too few arguments in function call

cudnn版本问题，解决：

添加判断，不同版本参数不同

#if CUDNN_VERSION_MIN(6, 0, 0)
  CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
      pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION,
      dataType<Dtype>::type));
#else
    CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
      pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION));
#endif