cuDNN error with simple code

vision
6

Thanks for the information.
I just created a conda environment with the same versions and couldn’t reproduce the error.
Which GPU are you using? The code runs fine on a GTX1080Ti.