CuDNNLSTM: UnknownError: не удалось найти реализацию dnn - PullRequest
Новая
       35

CuDNNLSTM: UnknownError: не удалось найти реализацию dnn

0 голосов
/ 01 февраля 2019

Я успешно запустил модель с LSTM в качестве первого слоя.Но из любопытства я заменяю LSTM на CuDNNLSTM.Но после model.fit он ответил следующее сообщение об ошибке: 

UnknownError: Fail to find the dnn implementation.
    [[{{node cu_dnnlstm_5/CudnnRNN}} = CudnnRNN[T=DT_FLOAT, _class=["loc:@training_2/Adam/gradients/cu_dnnlstm_5/CudnnRNN_grad/CudnnRNNBackprop"], direction="unidirectional", dropout=0, input_mode="linear_input", is_training=true, rnn_mode="lstm", seed=87654321, seed2=0, _device="/job:localhost/replica:0/task:0/device:GPU:0"](cu_dnnlstm_5/transpose, cu_dnnlstm_5/ExpandDims_1, cu_dnnlstm_5/ExpandDims_1, cu_dnnlstm_5/concat_1)]]
    [[{{node metrics_3/mean_squared_error/Mean_1/_1877}} = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_4852_metrics_3/mean_squared_error/Mean_1", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]()]]

Я попытался TestCudnnLSTM () в этом обсуждении и успешно прошел тест: 

Keras version: 2.2.4
Tensorflow version: 1.12.0
Creating Model
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
cu_dnnlstm_1 (CuDNNLSTM)     (None, 1000, 1)           16        
=================================================================
Total params: 16
Trainable params: 16
Non-trainable params: 0
_________________________________________________________________
None
Model compiled

Похоже, проблема возникает во время подбора модели.Но я не знаю точно, в чем проблема?

Ответы [ 3 ]

0 голосов
/ 28 февраля 2019

У меня была такая же проблема, когда я обновил тензор потока до 1.12.Ошибка была исправлена ​​после обновления версии CuDNN до 7,5 с 7. Я выполнил шаги, упомянутые в приведенном ниже URL-адресе, для обновления версии CuDNN (Примечание: шаги, упомянутые в ссылке, предназначены для установки CUDNN, но то же самое применимо и для обновления)

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

0 голосов
/ 05 августа 2019

Если вы получаете эту ошибку при установке Keras NN, укажите этот код при импорте 

from keras.backend.tensorflow_backend import set_session
import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
set_session(sess)

credit

0 голосов
/ 11 февраля 2019

Убедитесь, что у вас установлена ​​правильная версия драйвера Nvidia для той версии CUDA, которую вы используете.Вы можете проверить это здесь.https://docs.nvidia.com/deploy/cuda-compatibility/index.html#binary-compatibility

Я использую CUDA 9.0, но использовал драйвер Nvidia менее 384,81.Обновление драйвера Nvidia до нового исправило проблему для меня.

...