tensorflow出现 this is probably because cuDNN failed to initialize, so try looking to see if a……问题

报错信息
UnknownError: 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.

我的配置
tensorflow1.12
GTX1060
CUDA 10.0
cudnn 7.6.4

方法一:可靠粗暴
将tensorflow1.12替换为tensorflow1.8.0或者1.9.0
而且不能使用conda命令安装

pip install tensorflow-gpu==1.8.0

方法二:简单有效
建立Session时,将代码修改为

  config = tf.ConfigProto()
  config.gpu_options.allow_growth = True
  sess = tf.Session(config=config)

最好用方法二吧,毕竟现在tf2.0都出来了,降低版本不能长久~

全部评论

相关推荐

孤寡孤寡的牛牛很热情:为什么我2本9硕投了很多,都是简历或者挂,难道那个恶心人的测评真的得认真做吗
点赞 评论 收藏
分享
10-24 11:10
山西大学 Java
若梦难了:哥们,面试挂是很正常的。我大中厂终面挂,加起来快10次了,继续努力吧。
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务