cudnn is not being detected even after exporting the paths.
I tried the following:
My cudnn libraries are in ~/cuda/
export CUDNN_INCLUDE_DIR=~/cuda/include/
export CUDNN_LIB_DIR=~/cuda/lib64/

And then in python3, it returns nothing when I query for the cudnn version.
>>> import torch
>>> torch.backends.cudnn.version()
>>>