在支持GPU的工作站上运行RNN的Python代码时,发生以下错误:“InvalidArgumentError:没有注册OpKernel以支持使用这些attrs的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)]]`
所有导入版本都在我的系统中成功更新。我是一个noobie,非常感谢帮助,提前感谢。
如果你正确安装了GPU(检查命令“nvidia-smi”),那么确保你安装了正在使用的框架的gpu版本。如
sudo pip3 install tensorflow-gpu