I am using Ubuntu 18.04.4 with cuda 10.2.89 and cudnn 7.6.5 installed.
(base) root@3e5fbe29c57d:/usr/local/cuda/lib64# cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 #define CUDNN_MAJOR 7 #define CUDNN_MINOR 6 #define CUDNN_PATCHLEVEL 5 -- #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL) #include "driver_types.h"
It is ok to use cuda as backend, but “org.deeplearning4j.nn.layers.convolution.ConvolutionLayer - cuDNN not found” message displayed when using ConvolutionLayer.
I tried several times for re-install cudnn (The installation of CuDNN is just copying some files), but the problem is still there. What can I do for next step? Thanks!