CudnnRNNParamsSize Error

Question

CudnnRNNParamsSize Error

mickyyu96 opened this issue 7 years ago · 4 comments

Hello,
When i use CPU, it works normally. But when i use GPU, i got this error:

InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'CudnnRNNParamsSize' with these attrs. Registered devices: [CPU], Registered kernels:
device='GPU'; T in [DT_DOUBLE]; S in [DT_INT32]
device='GPU'; T in [DT_FLOAT]; S in [DT_INT32]

Can someone help me solve this problem ?
Thanks.

Answer 1 · 2018-03-02T08:01:45.000Z

I encounter the same problem.

Answer 2 · 2018-03-02T08:07:14.000Z

I solve it.
I uninstall the tensorflow, and install tensorflow-gpu. it works.
therefore, the tensorflow==1.4.0 in requirements.txt should be modified to tensorflow-gpu==1.4.0

Answer 3 · 2018-03-02T08:18:44.000Z

Thanks for your solution. I'll update it in Readme.

Answer 4 · 2018-03-09T11:46:43.000Z

Must it be version 1.4.0 ?
or can i use version 1.4.1 ?