Невозможно использовать tenorflow.keras.layers.CuDNNLSTM или keras.layers.CuDNNLSTM в моей среде Colab.

Question

Когда я пытался использовать tenorflow.keras.layers.CuDNNLSTM или keras.layers.CuDNNLSTM, я получил следующую ошибку:

InvalidArgumentError: No OpKernel was registered to support Op 'CudnnRNN' used by {{node cu_dnnlstm/CudnnRNN}}with these attrs: [dropout=0, seed=0, T=DT_FLOAT, input_mode="linear_input", direction="unidirectional", rnn_mode="lstm", is_training=true, seed2=0]
Registered devices: [CPU, XLA_CPU]

Я использую размещенную среду выполнения, и я предполагаю, что она также поддерживает GPU, но я заметил, что сообщение об ошибке выше показывает, что нет GPU. Не уверен, в чем проблема, но любая подсказка будет оценена по достоинству

Answer 1

Вам необходимо явно запросить среду выполнения с поддержкой графического процессора.

В меню «Runtime» выберите «Изменить тип среды выполнения», затем выберите «GPU» в разделе «Аппаратный ускоритель»: