首页 > 解决方案 > 已加载运行时 CuDNN 库:8.0.5,但源代码编译为:8.1.0

问题描述

当我运行 model.fit_generator 代码以使用 CNN 模型训练图像时出现此错误。我不明白错误,我该怎么办?谁能帮我?这是完整的错误描述`Loaded runtime CuDNN library: 8.0.5, but the source was compiled with: 8.1.0. CuDNN 库需要具有匹配的主要版本和相等或更高的次要版本。如果使用二进制安装,请升级您的 CuDNN 库。如果从源代码构建,请确保在运行时加载的库与编译配置期间指定的版本兼容。

标签: tensorflow

解决方案


我遇到了同样的错误“tensorflow/stream_executor/cuda/cuda_dnn.cc:362] Loaded runtime CuDNN library: 8.0.5 but source was compiled with: 8.1.0.” 我通过降级 TensorFlow 版本解决了这个问题,这里说您使用的是与 google colab CuDNN 版本不兼容的新版本的 TensorFlow。我使用了 TensorFlow 2.4.0 以及 2.4.0 版本所需的所有依赖项。
这里说明了使用哪个版本的 TensorFlow 来实现 cudnn 兼容性,https: //www.tensorflow.org/install/source


推荐阅读