我在用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]
真管用!有人知道我为什么会有这个问题吗?
3条答案
按热度按时间o2rvlv0m1#
此问题可能是帮助https://github.com/tensorflow/tensorflow/issues/24828
试着检查你有哪些版本的cudnn和tensorflow。
00jrzges2#
我用
conda install tensorflow-gpu
解决了这个问题。它自动安装cuDNN 7。3.1,问题解决了。x33g5p2x3#
这可能是由cuDNN的安装引起的。如果更改版本不起作用,请尝试检查系统环境变量中是否正确设置了库路径。请尝试更新您的系统,然后重新安装。