CuDNNLSTM: UnknownError: Fail to find the dnn implementation

后端 未结 6 575
感情败类
感情败类 2021-01-02 03:55

I have run the model with LSTM as the first layer successfully. But out of curiosity, I replace LSTM with CuDNNLSTM. But after model.fit, it replied the following error mess

6条回答
  •  悲&欢浪女
    2021-01-02 04:33

    In tensorflow 2.0 i got the same error while running RNN LSTM model.The reason was due to lower version of my cuDNN.In the tensorflow gpu requirements page it was recommended to have

    cuDNN SDK >= 7.4.1.
    

    You can refer for more details in https://www.tensorflow.org/install/gpu

    Asked in Tensorflow Reddit forum

    https://www.reddit.com/r/tensorflow/comments/dxnnq2/i_am_getting_an_error_while_running_the_rnn_lstm/?utm_source=share&utm_medium=web2x

提交回复
热议问题