首先需要了解的是,使用tensorflow-gpu 能够加速深度学习的计算过程,减少运行时间,是tensorflow的巨大优势。
其次 使用tensorflow-gpu有一个独立显卡或者CUDA支持的显卡。
链接:https://developer.nvidia.com/zh-cn/cuda-gpus
然后,我们一共要安装三个东西,CUDA、cudnn和tensorflow-gpu,三者有一个对应关系,参考官方链接:https://tensorflow.google.cn/install/source_windows
了解到对应的三个工具的版本号,一定要下对应版本,cudnn可能会有后面第三位数据的不同,但不影响,只要前几位相同就可。
自行搜索相应的版本号下载,百度上就能搜到。
安装CUDA:https://developer.nvidia.com/cuda-11.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
以及cudnn。
下载好后将cudnn的zip解压,改名文件夹为cudnn+版本号,复制到cuda的安装目录:C:\Program Files\NVIDIA GPU