Hello  all,
I have been trying to set up libtorch on centos8 as instructed on the installation guide provided by Pytorch. I have cuDNN 8 with 440.33 drivers and my cmake( version 3.1) looks like -
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(TorchTest)

set(CMAKE_PREFIX_PATH "/home/atharva/libtorch/libtorch/share/cmake/")
find_package(Torch REQUIRED)

add_executable(TorchTest main.cpp)
target_link_libraries(TorchTest /home/atharva/libtorch/libtorch/lib)

However, I run into the following error -

-- Caffe2: CUDA detected: 10.2
-- Caffe2: CUDA nvcc is: /usr/local/cuda/bin/nvcc
-- Caffe2: CUDA toolkit directory: /usr/local/cuda
-- Caffe2: Header version is: 10.2
-- Found cuDNN: v?  (include: /usr/include, library: /usr/lib64/libcudnn.so)
CMake Error at /home/atharva/libtorch/libtorch/share/cmake/Caffe2/public/cuda.cmake:170 (message):
  PyTorch requires cuDNN 7 and above.
Call Stack (most recent call first):
  /home/atharva/libtorch/libtorch/share/cmake/Caffe2/Caffe2Config.cmake:88 (include)
  /home/atharva/libtorch/libtorch/share/cmake/Torch/TorchConfig.cmake:40 (find_package)
  CMakeLists.txt:5 (find_package)


-- Configuring incomplete, errors occurred!
See also "/home/atharva/CLionProjects/TorchTest/cmake-build-debug/CMakeFiles/CMakeOutput.log".
See also "/home/atharva/CLionProjects/TorchTest/cmake-build-debug/CMakeFiles/CMakeError.log".

I have just started developing in c++ so this might be a very basic question…
What could I be doing wrong,
TIA