Windows中MatConvNet的GPU配置

为了读Zhang Lei老师实验室的代码,装上了MatConvNet,直接便宜CPU版本很简单,但是GPU略麻烦。

因为安装Caffe和TensorFlow的时候,都可以直接使用CUDA 8,但是Matlab对CUDA的支持比较慢,在2017a中才支持,所以要先下载2017a。

然后就简单了,直接运行下面函数:

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

注意的是,把cudaRoot换成你自己的,但是windows下,默认这个路径。
cudnnRoot也要换成自己的,这本身下载下来是一个压缩文件,所以放在那边是你自己决定。

posted on 2017-06-28 15:13  tendence  阅读(273)  评论(0编辑  收藏  举报