安装过tensorflow-gpu版本所必须

安装过tensorflow-gpu版本所必须

先去tensorflow官网查看对应的tensorflow版本所需的cuDNN和CUDA版本要求
windows
linux或macOS
在网站最下面就可查看(下面图片是windows的tensorflow各版本的需求)
安装过tensorflow-gpu版本所必须
再查看自家电脑支持什么版本CUDA
(路径:NVIDIA控制面板/帮助/系统信息/组件)
查看NVCUDA.DLL这一行后面的数字(下面图片中表示的最高支持11.0版本的CUDA)
安装过tensorflow-gpu版本所必须
最后去CUDAcuDNN官方下载网站下载对应版本即可
(注意:cuDNN的版本下载需要根据CUDA的版本对应下载!)