SeokjuLee / VPGNet

VPGNet: Vanishing Point Guided Network for Lane and Road Marking Detection and Recognition (ICCV 2017)
MIT License
487 stars 166 forks source link

cuDNN compile error: too few arguments to function cudnnSetPooling2dDescriptor #7

Closed vivounicorn closed 6 years ago

vivounicorn commented 6 years ago

@SeokjuLee Thanks for sharing this awesome work. I have run the caffe version on a Tesla M40 GPU,CUDA version 8.0,cuDNN version v5. when compile it,there is an error: ./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)); solution: 1、Because the cuDNN head file is too old,so replace the file ./include/caffe/util/cudnn.hpp with /usr/local/cuda/include/cudnn.h 2、 editing "caffe/cmake/Dependencies.cmake", at line 29 replace list(APPEND Caffe_LINKER_LIBS ${HDF5_LIBRARIES}) with list(APPEND Caffe_LINKER_LIBS ${HDF5_LIBRARIES} ${HDF5_HL_LIBRARIES})

Laulian commented 6 years ago

@vivounicorn Did you use the cudnn?I think you should comment USECUDNN=1 in your makefile.config file.

vivounicorn commented 6 years ago

@ln-scau thank you very much,I know if I comment USE_CUDNN := 1 it will work but i want to use cuDNN for acceleration.

daixiaogang commented 6 years ago

@vivounicorn ,the author claimed that they did not use cudnn in their code,only GPU.

vivounicorn commented 6 years ago

@daixiaogang thank you very much. I will implement a version with cuDNN.