如何使用 conda 为 tensorflow 安装兼容版本的 cudnn

Question

我尝试使用 tensorflow-gpu=1.6.0 在 gpu 上运行 python 代码 我已经使用 conda 命令安装了 tensorflow，假设安装了所有必需的库 当我运行代码时，出现以下错误：

<块引用>

加载的运行时 CuDNN 库：7605（兼容版本 7600）但源代码是用 7102（兼容版本 7100）编译的。如果使用二进制安装，请升级您的 CuDNN 库以匹配。如果从源代码构建，请确保在运行时加载的库与编译配置期间指定的兼容版本相匹配。 2021-02-16 02:29:34.892462: F tensorflow/core/kernels/conv_ops.cc:717] 检查失败：stream->parent()->GetConvolveAlgorithms(conv_parameters.ShouldIncludeWinogradNonfusedAlgo(), &alithms

任何人都可以帮助我如何使用 conda 查找和安装兼容版本以解决此问题...提前致谢

Answer 1

此页面列出了 tensorflow/cudnn 兼容性： https://www.tensorflow.org/install/source#linux

似乎您需要安装 cudnn 7 而不是 7.6 才能使用 tf 1.6.0