众所周知,当前主流深度学习的实现中调用的底层API都是cudnn,自己做项目需要开发深度学习模块时,也需要调用cudnn库,因此熟悉cudnn库是很有必要的。
从NVIDIA官方网站可以申请到CUDNN库与DEMO,通过研究DEMO,我们可以更好地使用CUDNN库。
下载CUDNN V4与DEMO
下载页面
需要注册一个开发者账户,流程很简单,这里不再赘述。
建议至少下载4个文件,win、linux、sample、doc
cudnn-7.0-win-x64-v4.0-prod.zip
cudnn-7.0-linux-x64-v4.0-prod.tgz
cudnn-sample-v4.tgz
libcudnn4-doc_4.0.7_amd64.deb
配置CUDNN V4
CUDNN V4的配置方法很简单
- 解压
cudnn-7.0-win-x64-v4.0-prod.zip
- 把解压后的文件夹中的
bin
,lib/x64
,include
中的内容拷贝至C:Program FilesNVIDIA GPU Computing ToolkitCUDAv7.5
中的bin
,lib/x64
,include
三个子目录下即可
解压DEMO
解压cudnn-sample-v4
获得mnistCUDNN
更改CUDA版本
工程默认使用CUDA7.0,但我用的是CUDA7.5,所以要在属性表里做调整
- 用
notepad++
打开mnistCUDNN_vs2010.vcxproj
文件 - 键入
Ctrl+H
,替换所有CUDA 7.0
为CUDA 7.5
(共有2个) - 保存退出
配置minstCUDNN工程
工程默认运行环境是Release,X64
- 双击sln文件打开工程
- 把项目属性改为Release,X64
- 编译
如果想要在Debug,X64
模式下运行,需要手动拷贝FreeImagelibwin64ReleaseFreeImage64.dll
文件至x64Debug
目录下。