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

Es werden keine Dateien oder Tests genannt. Beginne bei Environment > Local (GPU) und reproduziere die Erkennung von cuDNN 7.6.4, nachdem du die DLL im angegebenen CUDA-Pfad installiert hast; verfolge die Prüfung und ihr Wiederholungsverhalten. Fertig ist es, wenn die installierte Abhängigkeit erkannt wird oder der Bildschirm die Prüfung erneut ausführen und den tatsächlichen Fehler melden kann.