本人使用Spder直接运行tensorflow代码时,报如下错误:
UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed
背景:百度查了后,大家都说是tensorflow版本有问题,需要降低版本等等类似的解决方法,于是本人去如下如图的位置看了下自己的使用环境:(在此之前查看了任务管理器中的GPU是否正常)
结果看了一会后,又去Spder代码页面重新运行了下代码,结果就可以正常跑了,下图是跑的过程:
疑问:不知道为何就又能跑了?原因以后再补上,要是有小伙伴们知道,麻烦留言介绍介绍,先在此谢谢各位了!
补充说明:另外本人最开始遇到这个问题的时候,有一类型的笔记说加入以下代码可以解决该问题,本人最开始试过此方法,确实有效果,在这里本人也补充上:
#放在代码顶部的导入包的位置
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
最后,希望自己的方法对大家有帮助,有错误的地方也请大家直接指出,谢谢大家