Windows下编译MatConvNet+CUDA7.5+MATLAB2015b+cudnn

因为要跑一个 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+MATLAB2015b+cudnn)