tensorflow - 如何使用 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(), &algorithms)

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

标签： tensorflowanaconda

Answer 1

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

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