在win10中安装caffe并配置MATLAB和Python接口(支持GPU加速)

笔记本电脑配置(14年的老机器)CPU i54200 显卡:GTX850m

1、软件准备:

vs2013

cuda8.0(官网下载,查看显卡是否支持cuda),cudnn v5.0(支持cuda8.0)

Python使用的是anaconda2(Python2.7)(anaconda3在编译caffe出错)

caffe(使用的微软的版本,不用自己下载第三方库)

2、

(1)首先安装cuda8.0,一路点Next即可,不要更改安装路径

(2)解压cudnn v5,解压得到一个名为cuda的文件夹


caffe解压得到caffe-master文件,博主在D盘下新建了一个caffe文件夹,存放这些文件

(3)安装anaconda2,博主更改了安装路径,安装在D:/anaconda2下,安装过程中注意要把anaconda的路径添加到系统环境变量之中,一定要勾选第一项(安装时未截图,百度了一张截图,2和3的安装一致)


3、编译caffe

首先打开caffe下的Windows文件,找到一个名为CommonSettings.props.example的文件,并copy一份放在Windows文件下,更改附件名为CommonSettings.props。


用vs2013打开该文件,进行修改配置文件



由于微软的caffe的更新,需要在

CommonSettings.props中第55行,IncludePath中加入以下地址

$(MatlabDir)\toolbox\distcomp\gpu\extern\include;

否则会出现

到此配置文件修改完成,下面进行编译

4、

用vs2013打开caffe.sln

首先设置libcaffe的属性

右键libcaffe,生成


大约10分钟之后,编译成功,这时caffe文件中便多出了NugetPackages文件(第三方库文件)


之后依次编译caffe,matcaffe,pycaffe。为了保险,我又右键解决方案,生成解决方案


此时caffe编译成功,但是要想使用,需要配置接口环境

MATLAB:

(1)将上步release生成的matcaffe文件夹加入matlab的路径中


其次还要将D:\caffe\caffe-master\Build\x64\Release加入path环境变量之中


重启MATLAB,输入测试代码:caffe.reset_all

python配置:

编译python支持库后,你还需要执行如下之一才能在python里使用:

  • 设置环境变量 PythonPath,值为 <caffe_root>\Build\x64\Release\pycaffe,或者
  • 复制目录 <caffe_root>\Build\x64\Release\pycaffe\caffe 到 <python_root>\lib\site-packages 目录

博主一般使用MATLAB,Python未测试,待之后测试后在更新Python部分。

2018.4.11更新

python测试,打开Windows的dos窗口

输出python,试着引用caffe :import caffe

会出现

ImportError: No module named google.protobuf.internal

则需要安装protobuf,使用pip进行安装,在dos窗口输入

pip install protobuf(注意:不要使用conda的命令,博主未安装成功)

安装成功后,在打开python试着引用caffe


猜你喜欢

转载自blog.csdn.net/qq_35296537/article/details/79883454