【Tensorflow】tensorflow Loaded runtime CuDNN library: 7.0.5 but source was compiled with: 7.1.4.

问题描述

我的系统环境是windows,tensorflow版本是1.12.0,conda list显示cudnn版本是7.6.5,但在运行代码的时候会抛出题目所示的错误

问题解决

亲测有用。1.10.0之后的版本有这个问题,降低版本到1.8.0就可以了。

pip install --upgrade tensorflow-gpu==1.8.0

参考:https://forums.developer.nvidia.com/t/failed-to-get-convolution-algorithm-this-is-probably-because-cudnn-failed-to-initialize/67147/4

你可能感兴趣的:(效率工具,python,计算机视觉,深度学习,机器学习)