如何在Windows环境下配置Caffe
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了如何在Windows环境下配置Caffe相关的知识,希望对你有一定的参考价值。
参考技术A 1.配置环境我在自己的笔记本配置的caffe,配置的环境为:Windows 7 64位 + cuda6.5 + Opencv2.49 +VS2013。假设在配置caffe之前,你已经准备好这些。本文中将给出一些编译好的依赖库,如果你也是用的Windows 7 64位+VS2013,可以直接使用。
2.准备依赖库
在Windows下配置caffe,一个很主要的问题就是依赖库的编译。不像在Ubuntu下那么方便,在Windows下,依赖库都需要使用vs2013进行编译才能使用。下面我将介绍caffe需要的依赖库(如果你也是win7 64位+VS2013,可以直接使用我提供的依赖库)。
2.1 boost
boost可以下载源码进行编译,也可以直接下载安装文件。我使用的是后者,方便、快捷。
我使用的是:boost_1.56_0-msvc-12.0-64.exe
注意下载适合你的配置环境的boost版本即可。
下载完毕,双击运行安装文件即可。
2.2 Glog+Gflag+Protobuf+LevelDB+HDF5+LMDB+Openblas
这一部分的很多都是谷歌的开源库,不容易下载(你懂的)。所以我使用的是Neil Z. SHAO‘s Blog
提供的编译好的。
下载完,解压得到3rdparty文件夹。在下一段将会用到。
3.建立caffe工程
准备好了caffe需要的依赖库和环境之后,下面就可以建立caffe的vs项目,进行编译了。
3.1 下载caffe源码
可以从caffe的github主页下载源码。
下载地址:Caffe’s GitHub
解压文件,假设caffe源码所在目录为CAFFE_ROOT。
3.2 准备项目需要的依赖库和系统环境变量
经过上一阶段的准备,caffe项目所需的依赖库都已经准备好。
1.首先设置系统环境变量(以我的为例):
CUDA_PATH_V6_5 安装好cuda6.5之后,会自动添加环境变量CUDA_PATH_V6_5
OPENCV_2_49 D:/Tools/opencv2.49/build/
BOOST_1_56 D:/Tools/boost_1_56_0
2.将3rdparty文件夹放到CAFFE_ROOT
3.3 用vs建立caffe项目
1.用VS2013在CAFFE_ROOT下建立 win32 console application,选择空项目。
将项目的平台由32位改为64位
windows下检验caffe是否配置正确
windows下检验caffe是否配置正确:(注:不考虑搭建caffe的编译环境,而是直接使用caffe官网提供的二进制文件)
windows版本源码以及二进制库文件下载地址:https://github.com/BVLC/caffe/tree/windows
#encoding: utf-8
import numpy as np;
import sys,os;
# 注意下面的目录结构,根据自己的实际情况修改即可
caffe_root=‘D:/env/Caffe/caffe/‘;
sys.path.insert(0, caffe_root + ‘python‘);
import caffe as ca; # 引入自己的caffe的python接口
print("######"); # python打印6个#
import matplotlib.pyplot as plt;
plt.plot([1,2,3,4], [1,2,3,4]);
plt.show(); # 画一条45°直线
ca.set_mode_gpu(); #设置使用GPU,ca.set_mode_cpu();
print ("finish set_mode_gpu.")
# 下面5行验证python目录下的接口可以正常引用
import caffe.io as cai;
import caffe.net_spec as can;
import caffe.classifier as cac;
import caffe.draw as cad; # 这个文件需要安装pydot组件
import caffe.coord_map as caco;
print ("all ok")
依赖安装说明:
使用anaconda4.2.0,即python3.5,实际上已经继承了numpy,scipy,sklearn等一干工具。
还缺少的依赖有:pydot, 但是在python3.5平台下建议使用pydotplus, 而不是pydot,否则会报错。
另外在安装pydotplus的基础上使用pip重新安装graphviz。
备注:对这里的检验来说,只要用pip安装graphviz即可。但是在有些情况(绝大多数情况下)下,应当独立安装graphviz工具才可以。下载地址:http://www.graphviz.org/Download_windows.php
以上是关于如何在Windows环境下配置Caffe的主要内容,如果未能解决你的问题,请参考以下文章