1

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:-

enter image description here

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.

Thunder
  • 459
  • 3
  • 14

1 Answers1

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.

Gerry P
  • 7,662
  • 3
  • 10
  • 20
  • Gerry, even if CUDA 11.1, CUDNN 8.0.4 is incompatible with TF 1.14, does this specific error popped-up due to that? The error indicates it is unable to locate the CUDNN file. If there would be an incompatibility issue then maybe an error will generate during compilation. What do you think about it? – Thunder Nov 04 '20 at 03:53