can you post the full log of your command python setup.py install
I think it is possible that there is a /usr/local/cuda/include/cudnn.h which is older version and it is taking precedence over your version in home directory. can you remove /usr/local/cuda/include/cudnn.h if possible.
and ‘/usr/local/cuda/lib64/libcudnn.so’ is linked to ‘libcudnn.so.5.1’
I am not a sudo user, is there any other way to handle it? Thank you.
It seems that current pytorch master is not compatible with CUDNN v5.1 (this instructions said it is OK with NVIDIA cuDNN v5.x or above), is it out of update?