标签:
众所周知,当前主流深度学习的实现中调用的底层API都是cudnn,自己做项目需要开发深度学习模块时,也需要调用cudnn库,因此熟悉cudnn库是很有必要的。
从NVIDIA官方网站可以申请到CUDNN库与DEMO,通过研究DEMO,我们可以更好地使用CUDNN库。
下载页面
需要注册一个开发者账户,流程很简单,这里不再赘述。
建议至少下载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-7.0-win-x64-v4.0-prod.zip
bin
,lib/x64
,include
中的内容拷贝至 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5
中的bin
,lib/x64
,include
三个子目录下即可解压cudnn-sample-v4\
获得mnistCUDNN
工程默认使用CUDA7.0,但我用的是CUDA7.5,所以要在属性表里做调整
notepad++
打开mnistCUDNN_vs2010.vcxproj
文件Ctrl+H
,替换所有CUDA 7.0
为CUDA 7.5
(共有2个)工程默认运行环境是Release,X64
如果想要在Debug,X64
模式下运行,需要手动拷贝FreeImage\lib\win64\Release\FreeImage64.dll
文件至x64\Debug
目录下。
标签:
原文地址:http://www.cnblogs.com/nwpuxuezha/p/5190387.html