Tensorflow unable to find cudnn.h

I know that there are plenty of threads on installing Tensorflow on the JTX1, but I have yet to find this issue. I install cuda 8 toolkit, along with cudnn5.1 on Jetpack 2.3.1. However, I am unable to find the cudnn.h file anywhere, or the path to the cudnn library. The exact TF error message is

Auto-Configuration Error: Cannot find cudnn.h at /usr/lib/aarch64-linux-gnu/inlcude/cudnn.h

Hi traw1234,

We’re investigating this issue, the status will be shared once issue got clarified.


I got it to work by downloading the cuDNN files from Nvidia Developer and putting them into the Cuda directory.