首页 > 代码库 > 【MatConvNet】配置GPU
【MatConvNet】配置GPU
参照大神的方法: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
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。