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

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

原创最新推荐文章于 2024-02-04 15:48:24 发布 · 2.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#cudnn #caffe

caffe 专栏收录该内容

1 篇文章

订阅专栏

博客记录了cudnn版本问题的错误日志，解决办法是添加判断，因不同版本参数不同。主要围绕cudnn版本问题展开信息技术相关的解决思路。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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