首先安装显卡驱动
显卡驱动不会对下面的安装产生影响
CUDA、cuDNN以及tensorflow-gpu版本对应关系
安装CUDA
首先在CUDA官网上下载所需要的CUDA版本:https://developer.nvidia.com/cuda-downloads
按照提示安装CUDA,记录安装路径,默认路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
安装cuDNN
在cuDNN官网下载CUDA版本所对应的cuDNN版本:https://developer.nvidia.com/cudnn (需要登陆,可以用微信登陆)
下载好之后直接解压文件
讲下列文件复制到CUDA相应的位置即可
Copy \cuda\bin\cudnn64_7.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUD