crfasrnn icon indicating copy to clipboard operation
crfasrnn copied to clipboard

Is CuDNN 5 supported?

Open AlexTS1980 opened this issue 7 years ago • 2 comments

Is CuDNN 5 supported? I installed vanilla caffe w/out any problem, but crf-rnn caffe keeps telling me

./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:124: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));

AlexTS1980 avatar Feb 08 '17 17:02 AlexTS1980

Yes, you should check https://github.com/BVLC/caffe/pull/4159 I think I have merged this with a branch in caffe-crfasrnn. To use this in the default branch, you probably have to do some changes. Feel free to submit a PR.

bittnt avatar Mar 27 '17 03:03 bittnt

What changes do I need to make?

AlexTS1980 avatar Mar 28 '17 09:03 AlexTS1980