小编Jaw*_*ron的帖子

InvalidArgumentError: 没有注册 OpKernel 来支持 Op 'CudnnRNN'

我按照这里提供的步骤:https : //www.tensorflow.org/install/gpu 安装适用于我的 gpu 的 tensorflow,并使用 cuDNN SDK 但由于某些原因我仍然无法使用它,并且仍然得到这个错误:

tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'CudnnRNN' used by node sequential/cu_dnnlstm/CudnnRNN (defined at cudrnn.py:27) with these attrs: [input_mode="linear_input", T=DT_FLOAT, direction="unidirectional", rnn_mode="lstm", seed2=0, is_training=true, seed=0, dropout=0]
Registered devices: [CPU, XLA_CPU, XLA_GPU]
Registered kernels:
  device='GPU'; T in [DT_DOUBLE]
  device='GPU'; T in [DT_FLOAT]
  device='GPU'; T in [DT_HALF]
Run Code Online (Sandbox Code Playgroud)

即使 GPU 已注册,但 tensorflow 声称他们找不到 OpKernel。

我在用:

tensoflow-GPU 1.14.0

CUDA 10.0

Ubuntu 18.04

PS:我可以使用 tensorflow 运行其他代码,通过检查 nvidia-smi 我可以看到 GPU 正在被实际使用。

任何人都可以帮我知道为什么我会遇到这个问题吗?

这是我试图运行的代码:

model …
Run Code Online (Sandbox Code Playgroud)

python gpu python-3.x tensorflow

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

标签 统计

gpu ×1

python ×1

python-3.x ×1

tensorflow ×1