Keras-Tutorial icon indicating copy to clipboard operation
Keras-Tutorial copied to clipboard

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

Open renxingkai opened this issue 6 years ago • 2 comments

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)]]

renxingkai avatar Dec 06 '18 13:12 renxingkai

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

ningshixian avatar Dec 07 '18 02:12 ningshixian

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

renxingkai avatar Dec 07 '18 02:12 renxingkai