ningshixian/Keras-Tutorial

您好，我运行attentionLSTM代码报如下错误，想咨询下原因

renxingkai opened this issue · 2 comments

renxingkai commented

InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'CudnnRNN' with these attrs. Registered devices: [CPU], Registered kernels:

 [[Node: bilstm/CudnnRNN = CudnnRNN[T=DT_FLOAT, direction="unidirectional", dropout=0, input_mode="linear_input", is_training=true, rnn_mode="lstm", seed=87654321, seed2=0](bilstm/transpose, bilstm/ExpandDims_1, bilstm/ExpandDims_2, bilstm/concat)]]
ningshixian commented

CudnnRNN需要在GPU环境下才能跑,可以改为RNN/LSTM/GRU在CPU环境下运行

renxingkai commented

非常感谢您，我是keras新手，今天调通了，谢谢您的解答