安装tensorflow-GPU时,无法确定自己电脑需要安装哪个版本,这是在官网查询对应的版本即可
1.查看cuda的版本:
Win+R输入cmd——nvcc -V
输入"nvcc -V
"
可以看到cuda的版本为10.2
2.查看cudnn的版本:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include
在此路径下找到“cudnn.h
”文件
右键,选择以记事本
打开
可以看到cudnn的版本为7.6.5
3.进入tensorflow官网查看对应的版本:
https://tensorflow.google.cn/install/source_windows?hl=zh-cn#cpu