win10+vs2013+CUDA7.5 下搭建caffe

来源:互联网 发布:python oracle异常处理 编辑:程序博客网 时间:2024/06/05 17:52

一、准备

在此,首先要感谢happynear大神提供的包,GitHub上还有很多其它巨巨提供的资源可供下载,我这里采用的是happynear的,GitHub如下:
https://github.com/happynear/caffe-windows
下面是附加的第三方库,也打包好了:
https://pan.baidu.com/s/1bSzvKa

首先确定一点:所以的路径都是英文,包括计算机名,按住win+R,输入cmd,在这里可以看到一个路径C:\users\XXX 这里的XXX必须是英文,如果不是的话请先转换,否则后续可能导致一些编码或其它的未知错误
切记!!!

将GitHub的压缩包解压,再将第三方库解压并将其中的文件夹复制到3rdparty文件夹下
之后,将3rdparty下的bin添加到环境变量PATH中
注意先复制再添加环境变量

最后是CUDA,我现在(2016.8.22)下载的caffe中默认支持的是CUDA7.5,建议大家也下载这个版本,如果是其它版本,就需要将buildVS2013\MSVC目录下的vcproj文件用notepad或其它编辑器打开,找到里面CUDA的版本,将其改成自己的即可,否则打开工程会报错

二、编译

1、双击./src/caffe/proto/extract_proto.bat批处理文件来生成caffe.pb.h和caffe.pb.cc两个c++文件,和caffe_pb2.py这个python使用的文件。
2、打开./buildVS2013/MainBuilder.sln,打开之后切换编译模式至Release X64模式。如果你的显卡比较老或者没有显卡,请使用./build_cpu_only/MainBuilder.sln。 (很慢很慢很慢、说三遍)
3、点击上边绿色三角箭头运行即可

三、测试

http://pan.baidu.com/s/1mgl9ndu 下载已经转换好的MNIST的leveldb数据文件,解压至./examples/mnist文件夹中,然后运行根目录下的run_mnist.bat即可开始训练,训练日志会保存在./log文件夹中,以INFO开头,txt格式的日志文件中。

基本上都是happynear大神博客里的东西,加了一点自己遇到的问题
附上博客地址:http://blog.csdn.net/happynear/article/details/45372231/

0 0