error: too few arguments to function ‘cudnnStatus_t cudnnSetConvolution2dDescriptor(cudnnConvo问题解决

在使用cudnn编译针对OpenPose模型的Caffe时遇到错误:too few arguments to function 'cudnnSetConvolution2dDescriptor'。错误源于训练OpenPose的Caffe版本与系统中的cudnn版本不匹配。解决方案是将已成功编译的Caffe中的cudnn.hpp文件复制到问题Caffe目录,并重新编译。这可能由于OpenPose Caffe基于较旧的cudnn版本,而系统cudnn版本较高导致。
摘要由CSDN通过智能技术生成

在编译caffe时,如果你要使用cudnn,那么需要在caffe的Makefile.config文件中将CUDNN=1这一行取消注释,然后进行编译。

但是我在编译用于训练openpose模型的caffe时,遇到了如下错误:

In file included from ./include/caffe/util/device_alternate.hpp:40:0,
                 from ./include/caffe/common.hpp:19,
                 from src/caffe/syncedmem.cpp:1:
./include/caffe/util/cudnn.hpp: In function ‘void caffe::cudnn::setConvolutionDesc(cudnnConvolutionStruct**, cudnnTensorDescriptor_t, cudnnFilterDescriptor_t, int, int, int, int)’:
./include/caffe/util/cudnn.hpp:112:3: error: too few arguments to function ‘cudnnStatus_t cudnnSetConvolution2dDescriptor(cudnnConvolutionDescriptor_t, int, int, int, int, int, int, cudnnConvolutionMode_t, cudnnDataType_t)’
   CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
   ^
In file included from ./include/caffe/util/cudnn.hpp:5:0,
                 from ./include/caffe/util/device_alternate.hpp:40,
             

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值