CUDNN是用来实现高性能GUP加速的,快3到5倍的时间
首先通过NVIDIA控制面板查看电脑适用的CUDA版本
桌面鼠标右键即可:
CUDA和CUDNN的版本要对应,即:CUDA 9.1,CUDNN也下载9.1
安装CUDA
下载其他版本,请选择右下角Legacy Releases
下载CUDNN (最好用10.2以下)
点击下载,没有注册的话,这里需要注册一下。
下载之后双击CUDA安装包进行安装即可。
此目录是临时解压目录,安装完成后会自动删除,最终cuda还是会默认安装在C盘。所以安装在C盘即可
建议选择自定义
VS Intergration选择可能会安装失败,可以后续单独安装
单击下一步安装即可
安装完成后,管理员窗口cmd输入 nvcc --version 查看CUDA版本
CUDNN配置
cudnn是一个压缩文件不需要安装,只需要解压在自定义盘中即可
解压后打开CUDNN文件夹
进入cuda\bin找到cudnn64_7复制粘贴到 CUDA\bin
进入include找到cudnn.h 复制粘贴到 CUDA\include
lib\×64找到cudnn.lib 复制粘贴到 CUDA\lib\×64
CUDA的安装目录在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1
配置环境变量
CUDA9.1版本安装之后会自动配置好环境变量,不需手动添加
其他版本环境变量如有遗漏可按照此图进行添加
注意版本号要按照自己下载的版本写
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\libnvvp
完成!
最后再管理员窗口cmd输入 nvidia-smi 可查看当前GPU的使用情况
如出现不是内部命令则需要在系统变量path中新建环境变量,我的地址如下:
出现此页面即表示CUDA安装成功!!!