我按照这里提供的步骤: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)