CuDNNLSTM: UnknownError: Fail to find the dnn implementation

后端 未结 6 544
感情败类
感情败类 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:48

    I had the same issue , when I updated tensorflow to 1.12. Error got resolved after updating my CuDNN verstion to 7.5 from 7. I followed the steps mentioned in the below url for updating the CuDNN version (Note: The steps mentioned in the link are for installing CUDNN , but the same is applicable for update as well)

    https://jhui.github.io/2017/09/07/AWS-P2-CUDA-CuDNN-TensorFlow/

提交回复
热议问题