编译GPU版本Matconvnet

2017-04-28  本文已影响931人  学而时习之_不亦说乎

matconvnet是在Matlab(我是2015Rb版本)上实现神经网络计算的一个工具包,相对于其他的一些工具库,它的安装非常简单,昨天听友人说这个很方便,今天趁没事安转一个试试


Windows版本

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

cudnn/bin下面的cudnn64_5.dll复制到mex文件夹下。这一步,这个博客说的很详细

Train Example 1

Linux版本

vl_compilenn('enableGpu',true,'enableCudnn',true,'cudnnRoot','local/cuda')
#不用制定CUDA的路径,因为系统自己能找到

然后在Linux里面利用NVDIA自带的工具查看GPU的使用情况


Train Example 2
上一篇 下一篇

猜你喜欢

热点阅读