安装好 CUDA 后安装 cudnn
下载地址:cuDNN Download
需要注册信息,然后找CUDA对应的版本下载,所以要记住CUDA是哪个版本,我是 10.1 就选择第一项。
下载下来是个压缩包,解压后,里面有三个文件夹。
找到 CUDA 的安装路径,我的是 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1
配置:
- 复制 cuDNN bin 目录下的文件到 CUDA 的 bin 目录下(.dll)
- 复制 cuDNN include 目录下的文件到 CUDA 的 include 目录下(.h)
- 复制 cuDNN lib/x64 目录下的文件到 CUDA 的 lib/x64 目录下(.lib)
- 添加环境变量,把 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1libx64 加到path 中
不懂的看下面的图——