【MatConvNet】配置GPU

标签：blog   dll   技术   log   cpu   local   tco   .dll   eth   

参照大神的方法：http://www.th7.cn/system/win/201603/155182.shtml

第一步：需要安装cuda、VS2013；cuda默认路径，注意cuda版本和GPU要匹配

第二步：.下载cudnn，在matconvnet文件夹下建一个local文件夹，然后把cudnn放进去 （我改了文件名称为cudnn）

如图：

第三步：打开vl_compilenn.m，运行，等待编译结束

如图：

 
第四步把bin下的cudnn64_4.dll再复制到mex的文件夹下。

第五步：复制以下code到matlab命令窗，更改相应路径和文件夹名称，运行，等待编译结束。

vl_compilenn(‘enableGpu‘,true,‘cudaRoot‘,‘C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v8.0‘,‘cudaMethod‘ ,‘nvcc‘,‘enableCudnn‘,‘true‘,‘cudnnRoot‘,‘local/cudnn‘)

第六步：更改选项，opts.train.gpus= [1];   表示用GPU，[ ] 表示用CPU，千万别[0]

最后运行即可。

GPU和CPU对比

GPU：GTX 960M 

CPU: I5 4200 HQ

【MatConvNet】配置GPU

标签：blog   dll   技术   log   cpu   local   tco   .dll   eth   

原文地址：http://www.cnblogs.com/TensorSense/p/6284753.html