您的位置:首页 > 大数据 > 人工智能

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

2020-05-10 04:14 1166 查看

报错信息
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都出来了,降低版本不能长久~

小松与蘑菇 原创文章 126获赞 414访问量 12万+ 关注 私信
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐