在支持GPU的工作站上运行RNN的Python代码时,发生以下错误: ` InvalidArgumentError:没有使用这些attrs注册任何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)]]`
所有导入版本均已在我的系统中成功更新。 我是一名菜鸟友,感谢您的帮助。
答案 0 :(得分:1)
如果您已正确安装GPU(检查命令“ nvidia-smi”),请确保已安装了所用框架的gpu版本。例如
sudo pip3 install tensorflow-gpu