shows "cuDNN 7.6.4 is missing" even after installing cuDNN

**System Information:**
Visual studio Community Version 17.9.6
Extension Version: 17.18.2.2415501

**Describe the bug**

I am trying to train a model with the local GPU.
I go to Environment > Local (GPU)
I see the error in the attached screenshot saying "cuDNN 7.6.4 is missing"

I have tried the following:
- Clicked on Install cuDNN and follow the links to install cuDNN
- Tried getting exactly version cuDNN 7.6.4 from NVIDIA archives and copying the DLL to "c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin" as instructed
- Restarted the PC
- Reinstalled the extension

It still gives me the same error.

**Expected behavior**
It would be great if there was a "retry" button on this screen. It would be really helpful to rule out possible issues.

**Screenshots**