由于最近在学卷积神经网络,而且对matlab熟悉一些,所以就选用matlab来作为学习卷积神经网络的工具。首先本文假设你的机器已经安装好了cuda,cudnn,vs2015。本文参考了博客https://blog.csdn.net/u011935830/article/details/82187321。
1.MatConvNet的下载地址:https://github.com/vlfeat/matconvnet
2.下载之后将压缩包解压到一个文件夹,并在这个文件夹的matlab文件夹下新建local文件夹,把cudnn放入这个文件夹,然后matlab的当前文件夹目录进入这个文件夹的matlab文件夹
3.在命令行窗口输入:
mex -setup
4.如上图点击选择画红框部分
5.在命令行窗口输入:
vl_compilenn('enableGpu', true,...
'cuda