Your installed Caffe2 version uses cuDNN but I cannot find the cuDNN libraries. Please set the proper cuDNN prefixes and / or install cuDNN？

dalalaa: <p>Libtorch 1.5.0 can not find the cudnn installed in my computer. I gt the following error using libtorch_v1.5.0_cu92:</p>
<pre><code class="lang-auto">-- The C compiler identification is GNU 7.5.0
-- The CXX compiler identification is GNU 7.5.0
-- Check for working C compiler: /usr/bin/cc
-- Check for working C compiler: /usr/bin/cc -- works
-- Detecting C compiler ABI info
-- Detecting C compiler ABI info - done
-- Detecting C compile features
-- Detecting C compile features - done
-- Check for working CXX compiler: /usr/bin/c++
-- Check for working CXX compiler: /usr/bin/c++ -- works
-- Detecting CXX compiler ABI info
-- Detecting CXX compiler ABI info - done
-- Detecting CXX compile features
-- Detecting CXX compile features - done
-- Looking for pthread.h
-- Looking for pthread.h - found
-- Looking for pthread_create
-- Looking for pthread_create - not found
-- Looking for pthread_create in pthreads
-- Looking for pthread_create in pthreads - not found
-- Looking for pthread_create in pthread
-- Looking for pthread_create in pthread - found
-- Found Threads: TRUE  
-- Found CUDA: /usr (found suitable exact version "9.1") 
-- Found OpenCV: /media/zjk/tmp/library/opencv-3.4.1/build (found version "3.4.1") 
-- Found CUDA: /usr (found version "9.1") 
-- Caffe2: CUDA detected: 9.1
-- Caffe2: CUDA nvcc is: /usr/bin/nvcc
-- Caffe2: CUDA toolkit directory: /usr
-- Caffe2: Header version is: 9.1
-- Could NOT find CUDNN (missing: CUDNN_LIBRARY_PATH) 
CMake Warning at /media/zjk/tmp/library/temp/libtorch_cu92_v150/libtorch/share/cmake/Caffe2/public/cuda.cmake:109 (message):
  Caffe2: Cannot find cuDNN library.  Turning the option off
Call Stack (most recent call first):
  /media/zjk/tmp/library/temp/libtorch_cu92_v150/libtorch/share/cmake/Caffe2/Caffe2Config.cmake:88 (include)
  /media/zjk/tmp/library/temp/libtorch_cu92_v150/libtorch/share/cmake/Torch/TorchConfig.cmake:40 (find_package)
  CMakeLists.txt:15 (find_package)


-- Autodetected CUDA architecture(s):  6.1 6.1
-- Added CUDA NVCC flags for: -gencode;arch=compute_61,code=sm_61
CMake Error at /media/zjk/tmp/library/temp/libtorch_cu92_v150/libtorch/share/cmake/Caffe2/Caffe2Config.cmake:96 (message):
  Your installed Caffe2 version uses cuDNN but I cannot find the cuDNN
  libraries.  Please set the proper cuDNN prefixes and / or install cuDNN.
Call Stack (most recent call first):
  /media/zjk/tmp/library/temp/libtorch_cu92_v150/libtorch/share/cmake/Torch/TorchConfig.cmake:40 (find_package)
  CMakeLists.txt:15 (find_package)
</code></pre>
<p>Here’s my CMakeLists.txt:</p>
<pre><code class="lang-auto">cmake_minimum_required(VERSION 3.4)
project(ocr)
set(OpenCV_DIR /media/zjk/tmp/library/opencv-3.4.1/build)
find_package(OpenCV REQUIRED)
set(Torch_DIR /media/zjk/tmp/library/temp/libtorch_cu92_v150/libtorch/share/cmake/Torch) #GPU 1.5.0
find_package(Torch REQUIRED)
add_executable(ocr inference.cpp model.cpp utils.cpp)
target_link_libraries(ocr ${TORCH_LIBRARIES} ${OpenCV_LIBS}) # ${CMAKE_THREAD_LIBS_INIT})
</code></pre>
<p>And the environment variables about cuda and cudnn in ~/.bashrc is as follows:</p>
<pre><code class="lang-auto">export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/dai/cuda9/lib64/
export PATH=$PATH:/home/dai/cuda9/bin
export CUDA_HOME=$CUDA_HOME:/home/dai/cuda9
export ORACLE_HOME=/home/dai/library/instantclient_11_2
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$ORACLE_HOME
export CUDNN_INCLUDE_DIR=/media/zjk/tmp/library/temp/cuda/include/
export CUDNN_LIB_DIR=/media/zjk/tmp/library/temp/cuda/lib64/
export CUDNN_PATH=/media/zjk/tmp/library/temp/cuda/lib64/libcudnn.so.7
export CUDNN_LIBRARY=/media/zjk/tmp/library/temp/cuda/lib64/libcudnn.so
</code></pre>
<p>The project works fine when I built with libtorch_v1.0.1_cu92.<br>
How can I solve this problem? Looking foward to your reply.</p>

dalalaa: <p>I solved this problem by replace<br>
<code>export CUDNN_LIBRARY=/media/zjk/tmp/library/temp/cuda/lib64/libcudnn.so</code><br>
by<br>
<code>export CUDNN_LIBRARY=/media/zjk/tmp/library/temp/cuda/lib64/</code></p>