Pytorch only support cuDNN 6.x or above, but another program on my computer needs cuDNN 5.1. So there are 2 versions on my computer and paths are:
/usr/local/cuda-8.0/lib64/cuDNNv5
/usr/local/cuda-8.0/lib64/cuDNNv6
However, even I add /usr/local/cuda-8.0/lib64/cuDNNv6 in LD_LIBRARY_PATH and CUDNN_LIB_DIR, cudnn is not found when I install pytorch.
Any one can give some help? How to specify CUDNN path for installing pytorch.
Thanks