keras Tensorflow cuDNN兼容性

b4qexyjb  于 2023-04-30  发布在  其他
关注(0)|答案(3)|浏览(123)

我在用Keras做模型。
在编译时,我的模型不工作,并弹出一条错误消息:tensorflow.python.framework.errors_impl.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.
我的电脑规格如下:
GPU:RTX 2070,Tensorflow版本:1.13.1,Python版本:3.6.5,CUDA:10.0,cuDNN:7.4.2
我试过cuDNN 7。5.0和这个链接:cannot train Keras convolution network on GPU但更改cuDNN版本对我不起作用。
我试了这些代码:

>>>import tensorflow as tf
>>>a = tf.constant([1])
>>>b = tf.constnat([2])
>>>sess = tf.Session()
>>>with tf.device('/gpu:0'):
...    print(sess.run(a+b))
...
[3]

真管用!有人知道我为什么会有这个问题吗?

o2rvlv0m

o2rvlv0m1#

此问题可能是帮助https://github.com/tensorflow/tensorflow/issues/24828
试着检查你有哪些版本的cudnn和tensorflow。

00jrzges

00jrzges2#

我用conda install tensorflow-gpu解决了这个问题。它自动安装cuDNN 7。3.1,问题解决了。

x33g5p2x

x33g5p2x3#

这可能是由cuDNN的安装引起的。如果更改版本不起作用,请尝试检查系统环境变量中是否正确设置了库路径。请尝试更新您的系统,然后重新安装。

相关问题