Closed vivounicorn closed 6 years ago
@vivounicorn Did you use the cudnn?I think you should comment USECUDNN=1 in your makefile.config file.
@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.
@vivounicorn ,the author claimed that they did not use cudnn in their code,only GPU.
@daixiaogang thank you very much. I will implement a version with cuDNN.
@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})