美文网首页
Windows下编译MatConvNet+CUDA7.5+MAT

Windows下编译MatConvNet+CUDA7.5+MAT

作者: 凉稀饭豆腐脑 | 来源:发表于2016-11-23 17:00 被阅读0次

    因为要跑一个 João F. Henriques大神他们的Fully-Convolutional Siamese Networks for Object Tracking跟踪算法,需要使用matlab+cuda等等,因此按照他们给出的方案进行配置,其中遇到了一些小问题,记录下来以备重装。

    MatConvNet概述

    MatConvNet是VLFeat的一部分,是卷积神经网络(CNN)的一个实现。既可以使用matlab编程,又能使用GPU加速,是一个不错的CNN库。
    [MatConvNet首页]

    安装

    下载MatConvNet工具箱

    • 点击下载工具箱
    • 我这里使用的版本是 matconvnet-1.0-beta20.tar

    解压文件

    • 解压到 matconvnet-1.0-beta20

    编译安装

    • 编译前确定matlab是否绑定好C++编译器了,可以使用mex -setup查看设置编译器,一般先安装Visual Studio再安装MATLAB就可以自动配置好了,我这里使用的VS2013和MATLAB2015b,并安装了对应的CUDA7.5,按照论文里的配置。
    • 打开MATLAB,路径切换至matconvnet-1.0-beta20
    • addpath matlab 将matlab文件夹添加到搜索路径
    • 使用vl_compilenn编译参数类似于下面,需要设置CUDA路径, cudnn的路径,我把cudnn放在matconvnet-1.0-beta20/local里面了。如果不加参数默认是CPU。

    vl_compilenn('enableGpu',true,'cudaRoot','C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v7.5', 'cudaMethod' ,'nvcc', 'enableCudnn','true', 'cudnnRoot','local/cuda')

    • 在MATLAB里输入命令vl_setupnn安装工具箱
    • 测试 vl_testnn

    这样就完成了

    相关文章

      网友评论

          本文标题:Windows下编译MatConvNet+CUDA7.5+MAT

          本文链接:https://www.haomeiwen.com/subject/cwevpttx.html