python-3.x - 在任何子目录中找不到任何匹配版本“8”的 cudnn.h

Question

出于某些特定目的，我在 Ubuntu 16.04 上使用 CUDA 11.1、CUDNN 8.0.4 和 TensorRT 7.2从源代码构建 TF 1.14，但出现错误：-

我已验证 CUDNN 已安装在 /usr/include/cudnn.h 并按照这个 stackoverflow 答案我已将 cudnn.h 文件复制到 /usr/local/cuda/ 和“libcudnn8_8.0.4.30-1+cuda11.1_amd64 .deb" , "libcudnn8-dev_8.0.4.30-1+cuda11.1_amd64.deb" 文件到 /usr/local/cuda。谁能帮帮我吗？

作为旁注，我认为这可能不是这个问题的原因；机器上安装了两个 CUDNN 和 CUDA 版本。

标签： python-3.xtensorflowcudnn

Answer 1

Robert、CUDA 11.1、CUDNN 8.0.4 与 TF 1.14 不兼容。首先，我建议您升级到 tensorflow 2，但这不是必需的。如果你安装了 Anaconda，我发现最好使用 Conda 安装 tensorflow。原因是 conda 会自动安装正确版本的 cuda 工具包和 cudnn。Pip 不这样做，您必须手动下载正确的版本并更改路径环境变量以指向您放置它们的目录等。所以首先我会卸载 tensorflow，然后用 conda 重新安装它。如果您需要更多信息，这里有一个很好的指南。