./include/caffe/util/cudnn.hpp: error: too few arguments to function ‘SetPooling2dDescriptor
生活随笔
收集整理的這篇文章主要介紹了
./include/caffe/util/cudnn.hpp: error: too few arguments to function ‘SetPooling2dDescriptor
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
https://blog.csdn.net/tianrolin/article/details/71246472
這是由于所使用的cuDNN版本不一致的導致的,作者配置環境是cuDNN 4.0,但是5.0版本后的cuDNN接口有所變化。
解決方法?:將以下幾個文件用最新BVLC版本的caffe對應文件替換并重新編譯
./include/caffe/util/cudnn.hpp ./include/caffe/layers/cudnn_conv_layer.hpp ./include/caffe/layers/cudnn_relu_layer.hpp ./include/caffe/layers/cudnn_sigmoid_layer.hpp ./include/caffe/layers/cudnn_tanh_layer.hpp./src/caffe/layers/cudnn_conv_layer.cpp ./src/caffe/layers/cudnn_conv_layer.cu ./src/caffe/layers/cudnn_relu_layer.cpp ./src/caffe/layers/cudnn_relu_layer.cu ./src/caffe/layers/cudnn_sigmoid_layer.cpp ./src/caffe/layers/cudnn_sigmoid_layer.cu ./src/caffe/layers/cudnn_tanh_layer.cpp ./src/caffe/layers/cudnn_tanh_layer.cu總結
以上是生活随笔為你收集整理的./include/caffe/util/cudnn.hpp: error: too few arguments to function ‘SetPooling2dDescriptor的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: NLP(自然语言处理) - Tricks
- 下一篇: 【扫盲】小白基础-SDN详解