Darknet compile with cudnn support

Developer Tools CUDA Developer Tools
1

Hello everyone, I am trying to train a network with yolov4 model on my custom object. To achieve that I found that darknet is a solid framework for this operation. I clone the https://github.com/AlexeyAB/darknet the repository. Change some parameters on makefile like GPU=1 , CUDNN=1. I got all my drivers installed and nvcc --version is 10.1 . After I run make in the directory I face with

chmod +x *.sh
g++ -std=c++11 -std=c++11 -Iinclude/ -I3rdparty/stb/include -DGPU -I/usr/local/cuda/include/ -DCUDNN -Wall -Wfatal-errors -Wno-unused-result -Wno-unknown-pragmas -fPIC -Ofast -DGPU -DCUDNN -I/usr/local/cudnn/include -c ./src/image_opencv.cpp -o obj/image_opencv.o
In file included from ./src/image.h:3:0,
                 from ./src/image_opencv.h:4,
                 from ./src/image_opencv.cpp:1:
include/darknet.h:46:10: fatal error: cudnn.h: No such file or directory
 #include <cudnn.h>
          ^~~~~~~~~
compilation terminated.
Makefile:174: recipe for target 'obj/image_opencv.o' failed
make: *** [obj/image_opencv.o] Error 1

what am i missing?

7 Likes