Hello bala (prasanna)

I would like to ask a question with respect to cuda version 9.0 and cudnn 7.5.1 Error is something same with respect to this one, so sharing my experience can be beneficial to all. I have successfully installed both of these and successfully configured cuda for both tensorflow and pytorch.

I am testing torch gpu on the attached code in the link.

I got the following error RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

I tried some way but I think this is something to do with cuda. cuda is working successful with torch.

Any ideas will be welcomed.

With Regards
Sanpreet Singh