daijifeng001 / caffe-mnc

Caffe branch for MNC

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Build failing Cuda 7.5 CUDNN v5.1

pbarker opened this issue · comments

Hey I was trying to build this on the current Deep Learning AMI on AWS, but the build is failing with

make: *** [.build_release/src/caffe/util/benchmark.o] Error 1
In file included from ./include/caffe/util/device_alternate.hpp:40:0,
                 from ./include/caffe/common.hpp:19,
                 from ./include/caffe/data_reader.hpp:8,
                 from src/caffe/util/blocking_queue.cpp:4:
./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:127:41: error: too few arguments to function ‘cudnnStatus_t cudnnSetPooling2dDescriptor(cudnnPoolingDescriptor_t, cudnnPoolingMode_t, cudnnNanPropagation_t, int, int, int, int, int, int)’

The deep learning AMI is running CUDA 7.5 and CUDNN v5.1 on amazon linux. It looks like this may be related to the version of CUDNN. Any idea on how to get up and running here?

commented

It maybe need CUDNN v4.

Same issue. I got it working with cuDNN v4, but then the actual output of the network was incorrect. I'm currently trying to see whether it's possible to merge Caffe's master into this (unofficial) fork.