Ошибка ввода неверного аргумента с RNN на GPU - PullRequest
0 голосов
/ 11 сентября 2018

Во время работы моего кода Python для RNN на рабочей станции с графическим процессором произошла следующая ошибка: ` InvalidArgumentError: Нет зарегистрированного OpKernel для поддержки Op 'CudnnRNN' с этими атрибутами.Зарегистрированные устройства: [CPU], Зарегистрированные ядра:

 [[Node: lstm_1/CudnnRNN = CudnnRNN[T=DT_FLOAT, direction="unidirectional", dropout=0, input_mode="linear_input", is_training=true, rnn_mode="gru", seed=87654321, seed2=0](lstm_1/transpose, lstm_1/ExpandDims_1, lstm_1/Const_1, lstm_1/concat)]]`

Все версии импорта успешно обновлены в моей системе.Я новичок, и помощь высоко ценится, заранее спасибо.

1 Ответ

0 голосов
/ 30 марта 2019

если у вас правильно установлен графический процессор (проверьте команду "nvidia-smi"), то убедитесь, что вы установили версию gpu используемой платформы.такие как

 sudo pip3 install tensorflow-gpu
...