Could not find any cudnn.h matching version '8' in any subdirectory

Question

For some specific purpose, I am building TF 1.14 from source with CUDA 11.1, CUDNN 8.0.4 & TensorRT 7.2 on Ubuntu 16.04 but getting an error:-

I have verified that CUDNN is installed at /usr/include/cudnn.h and following this stackoverflow answer I have copy pasted cudnn.h file at /usr/local/cuda/ and "libcudnn8_8.0.4.30-1+cuda11.1_amd64.deb" , "libcudnn8-dev_8.0.4.30-1+cuda11.1_amd64.deb" files to to /usr/local/cuda. Can anyone please help me?

As a side note, which I think probably not the cause of this issue; there are two CUDNN and CUDA versions installed on the machine.

Answer 1

Robert, CUDA 11.1, CUDNN 8.0.4 are not compatible with TF 1.14. First I would recommend you upgrade to tensorflow 2 but it is not necessary. I found it best to install tensorflow using Conda if you have Anaconda installed. Reason is conda will install the right versions of the cuda toolkit and cudnn automatically. Pip does not do that and you manually have to download the right versions and change the path environment variable to point to the directories where you placed them etc. So first I would uninstall tensorflow, then reinstall it with conda. If you need further information there is a good guide located here.