tensorflow.python.framework.errors_impl.UnknownError: Fail to find the dnn implementation. [Op:Cudnn

Posted 2021-12-25 沉迷单车的追风少年

项目场景：

lstm特征提取

---

问题描述：

Traceback (most recent call last):
  File "encode_lstm.py", line 62, in <module>
    output = encoder(inputs)
  File "/root/anaconda3/envs/tf35/lib/python3.7/site-packages/tensorflow/python/keras/engine/base_layer.py", line 985, in __call__
    outputs = call_fn(inputs, *args, **kwargs)
  File "encode_lstm.py", line 15, in call
    output = self.lstm(inputs)
  File "/root/anaconda3/envs/tf35/lib/python3.7/site-packages/tensorflow/python/keras/layers/recurrent.py", line 663, in __call__
    return super(RNN, self).__call__(inputs, **kwargs)
  File "/root/anaconda3/envs/tf35/lib/python3.7/site-packages/tensorflow/python/keras/engine/base_layer.py", line 985, in __call__
    outputs = call_fn(inputs, *args, **kwargs)
  File "/root/anaconda3/envs/tf35/lib/python3.7/s