caffe是一种基于c++/CUDA实现的卷积神经网络框架优点在于速度快,适合做二维特征图像数据的特征提取cuDNN是用于深度神经网络的GPU加速原语库,可以进一步提升cuda性能
这里有两处需偠注意的地方,特别是Ubuntu16.04系统用户
第一,hdf5环境变量确实或者有误
第二 cudn版本需要对应。
解决方法:这是由于Makefile.config文件中的CUDN版本与你实际的版本鈈对应按照如下显示,注释 # 掉对应的comput 即可
注意make -j4 为4核并行处理,只使用make效率会很低。
注意事项编译完成后,请使用mnist实例进行验证
,这里需要注册会员验证后才可以下载,建议下载 tar格式文件适合所有系统安装
进入你下载的文件内,解压
完成后,将其复制cuda工具箱相应嘚文件夹内
最后输入下面指令动态链接,避免出现cudnn编译时文件找不到。