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

问题描述 投票:0回答:1

在支持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,非常感谢帮助,提前感谢。

tensorflow keras deep-learning keras-layer
1个回答
1
投票

如果你正确安装了GPU(检查命令“nvidia-smi”),那么确保你安装了正在使用的框架的gpu版本。如

 sudo pip3 install tensorflow-gpu
© www.soinside.com 2019 - 2024. All rights reserved.