出现Failed to get convolution algorithm的解决方法

当运行卷积神经时出现了问题:Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

这一般就是cuda+cuDNN+TensorFlow的版本匹配不一致,当然大概率就是TensorFlow的版本过高了,降级即可:

sudo pip3 install --upgrade --force-reinstall tensorflow-gpu==1.9.0 --user

 

posted @ 2019-02-28 14:56  codeg  阅读(23389)  评论(0编辑  收藏  举报