When installing tensorflow-GPU, you can't determine which version you need to install on your computer. You can check the corresponding version on the official website.
1. Check the cuda version:
Win+R input cmd——nvcc -V
Enter " nvcc -V
"
You can see that the version of cuda is10.2
2. Check the version of cudnn:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include
cudnn.h
Right click on the " " file in this path
, select to 记事本
open ,
and you can see that the version of cudnn is7.6.5
3. Enter tensorflow official website to check the corresponding version:
https://tensorflow.google.cn/install/source_windows?hl=zh-cn#cpu