问题描述

我的系统环境是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

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐