1. 自己的运行环境
- NVIDIA GeForce 940MX
- Windows 10 64bit
- MATLAB 2016b
- Visual Studio 2015
- CUDA 8.0.61 for Win10
- MatConvNet https://github.com/vlfeat/matconvnet
2. 注意事项
CUDA似乎只在8.0版本开始才支持VS 2015,之前用CUDA 7.5出很多问题,后来换8.0才弄好,安装了VS 2015的同学需要注意此信息。其他配置项可以灵活处理,留意一下MatConvNet的文档就好。
MATLAB、VS2015、CUDA几乎都是默认安装,就本文的范围而言,没有手动环境变量要求。
3. 编译
S1. 首先确认MATLAB确实配置了VS2015作为C和C++编译器
mex -setup
mex -setup C++
然后根据MATLAB中的提示进行操作即可。
S2. 在MATLAB中进入下载的MatConvNet目录中的matlab文件夹,在命令行中运行
vl_compilenn('enableGpu', true, 'cudaRoot', 'D:/CUDA/8.0/Toolkit')
最后一个参数为CUDA的路径,一定要根据实际情况输入。编译时需要一段时间,中间可能会有一些乱七八糟的警告,不必理会,只