小编Ste*_*ev.的帖子

无法获得卷积算法。这可能是因为cuDNN无法初始化,

在Tensorflow / Keras中,从https://github.com/pierluigiferrari/ssd_keras运行代码时,使用估算器:ssd300_evaluation。我收到此错误。

无法获得卷积算法。这可能是因为cuDNN初始化失败,所以请尝试查看上面是否打印了警告日志消息。

这与未解决的问题非常相似:Google Colab错误:无法获得卷积算法。这可能是因为cuDNN初始化失败

我正在运行的问题:

的Python:3.6.4。

Tensorflow版本:1.12.0。

Keras版本:2.2.4。

CUDA:V10.0。

cuDNN:V7.4.1.5。

NVIDIA GeForce GTX 1080.

Also I ran:

import tensorflow as tf
with tf.device('/gpu:0'):
      a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
      b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
      c = tf.matmul(a, b)
with tf.Session() as sess:
print (sess.run(c))
Run Code Online (Sandbox Code Playgroud)

With no errors or issues.

The minimalist example is:

 from keras import backend as K
 from keras.models import …
Run Code Online (Sandbox Code Playgroud)

python keras tensorflow cudnn

10
推荐指数
5
解决办法
1万
查看次数

标签 统计

cudnn ×1

keras ×1

python ×1

tensorflow ×1