CuDNNLSTM: UnknownError: Fail to find the dnn implementation

后端 未结 6 600
感情败类
感情败类 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条回答
  •  萌比男神i
    2021-01-02 04:47

    I Installed tensorflow and keras using conda in the Virtual env and this solved it.

    conda install tensorflow
    conda install keras
    

提交回复
热议问题