zoukankan      html  css  js  c++  java
  • 【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

     

     

  • 相关阅读:
    单元测试-伪对象、桩对象、模拟对象
    单元测试那些事
    vscode 开发go版本1.14项目
    C#中的 async await
    一手遮天 Android
    一手遮天 Android
    一手遮天 Android
    一手遮天 Android
    一手遮天 Android
    一手遮天 Android
  • 原文地址:https://www.cnblogs.com/TensorSense/p/6284753.html
Copyright © 2011-2022 走看看