mini-caffe
mini-caffe copied to clipboard
error while compiling miniCaffe using cuDNN
I compile the minicaffe as below: mkdir build cd build cmake .. -DCMAKE_BUILD_TYPE=Release -DUSE_CUDA=ON -DUSE_CUDNN=ON make And I have already intsalled the cudnn.But a problem was occured: [ 0%] Building NVCC (Device) object CMakeFiles/cuda_compile.dir/src/layers/cudnn/cuda_compile_generated_cudnn_bn_layer.cu.o /home/lgz/mini-caffe/src/layers/cudnn/././cudnn.hpp(97): error: too few arguments in function call
1 error detected in the compilation of "/tmp/tmpxft_00003fb4_00000000-7_cudnn_bn_layer.cpp1.ii". CMake Error at cuda_compile_generated_cudnn_bn_layer.cu.o.cmake:266 (message): Error generating file /home/lgz/mini-caffe/build/CMakeFiles/cuda_compile.dir/src/layers/cudnn/./cuda_compile_generated_cudnn_bn_layer.cu.o
What's the matter ?
My cudnn is newest one and my OS is ubuntu 16.04.Maybe the version of cuDNN cause this problem ?
I find a post of NVIDIA which has the same question and the solution is work. https://devtalk.nvidia.com/default/topic/1002826/gpu-accelerated-libraries/question-about-cudnnsetconvolution2ddescriptor/
What's your cudnn version? I think the newest version is cudnn6, but I didn't test with it. Only cudnn5.1 is tested.
Adding the below will solver it. -DCUDA_NVCC_FLAGS='-std=c++11' eg: cmake .. -DCMAKE_BUILD_TYPE=Release -DUSE_CUDA=ON -DUSE_CUDNN=ON -DCUDA_NVCC_FLAGS='-std=c++11'