在GPU上使用RNN获取无效的参数错误

时间:2018-09-11 19:38:58

标签: tensorflow keras deep-learning keras-layer

在支持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)]]`

所有导入版本均已在我的系统中成功更新。 我是一名菜鸟友,感谢您的帮助。

1 个答案:

答案 0 :(得分:1)

如果您已正确安装GPU(检查命令“ nvidia-smi”),请确保已安装了所用框架的gpu版本。例如

 sudo pip3 install tensorflow-gpu