您的位置:首页 > 编程语言 > MATLAB

MATLAB 2016中搭建MatConvNet运行环境(调用GPU运行)

2018-01-03 18:30 911 查看

自己的运行环境

NVIDIA GeForce 940MX

Windows 10 64bit

MATLAB 2016b

Visual Studio 2015

CUDA 8.0.61 for Win10

MatConvNet https://github.com/vlfeat/matconvnet

注意

CUDA似乎只在8.0版本开始才支持VS 2015,之前用CUDA 7.5出很多问题,后来换8.0才弄好,安装了VS 2015的同学需要注意此信息。其他配置项可以灵活处理,留意一下MatConvNet的文档就好。

MATLAB、VS2015、CUDA几乎都是默认安装,就本文的范围而言,没有手动环境变量要求。

编译

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的路径,根据实际情况输入。编译时需要一段时间,中间可能会有一些乱七八糟的警告,不必理会,只要最后出现若干“mex已经成功完成”即可。

测试

在命令行中输入

run vl_setupnn


再输入

vl_testnn('gpu', true)


MATLAB会运行一段时间,最后会输出类似如下信息,说明环境搭建成功。

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  cuda matlab gpu 深度学习