下面进入正题。
1.依赖环境的配置
本人的系统Windows10 64位,版本是14393.321
首先需要下载并安装visual studio 2013(不要下vs2015)
下载NVIDIA CUDA7.5(最新的是8.0,但是目前git上的版本对应的都是7.5,需要修改vs的配置文件才能打开工程文件,能够正常编译,如果是gtx1000系列帕斯卡的新卡,请安装8.0)
下载NVIDIA CUDNN V5.0(同样,最新的是5.1,如果你是帕斯卡的显卡,那么请使用5.1)
以上的软件的参考地址https://developer.nvidia.com/cuda-gpus
然后你需要看一下你的GPU的计算能力是多少,用作深度学习的显卡计算能力至少是3.0。本人使用的GTX960的计算能力是5.0,其他显卡可以在上面的网站上找到对应的数值,如果你的显卡比较烂,可以编译一个纯CPU的版本(不过训练速度感人),建议还是去换设备23333
如果需要Python(推荐2.7)的话,建议下载Anaconda,它集成了常用的科学计算软件包。
前往github下载Windows版的caffe源码:https://github.com/BVLC/caffe/tree/windows
选择右上角的绿色按钮,选择下载zip,将zip文件解压,最好放在磁盘的根目录上(文件夹名称尽量不要带中文,免得出问题),需要保证磁盘的空余空间有6GB以上。
下载依赖libraries:找到 caffe源码目录\scripts\download_prebuilt_dependencies.py 并运行之。
> python scripts\download_prebuilt_dependencies.py
如果出错了你就打开那个py文件自己去下载,然后在caffe源码目录下建立libraries文件夹并解压进去。
2.Build
1.把CommonSettings.prop.example复制一份,并把文件名改为CommonSettings.prop
2.打开Windows文件夹,打开caffe.sln,待扫描完毕后,请确认生成的模式是Release|X64,然后右键项目生成解决方案,等待漫长的编译(大概半小时)。如果没有报错就恭喜你,可以去测试了。
3.如果出现报错请把caffe--属性--C/C++--常规---把警告视为错误 给改成否(/WX-)
4.如果编译过程中出现libcaffe中math的一个模块出现无法生成obj,请在vs报错列表中双击该错误打开文件,重新保存。
3.测试:
首先需要去下载cifar-10数据集
然后将其转化为lmdb格式,供caffe软件使用。
使用example里面的快速模式进行训练,大概只需要1分多钟就完成了,然后我们进行测试
这里是从百度上搞来的一只狗狗图片
然后我们把它丢进去进行测试
识别成功。
或许有些同学对显卡的性能存在一些疑惑,于是乎我又跑了一遍完整的训练。其中5万张做训练,1万张是测试集
跑完总共花了将近1个小时,准确度是0.786
显卡最高的时候是83°C
我的电脑的配置是i7 4720HQ+GTX960M+16G RAM,大家参考一下吧
以下正在施工。。。。
Comments 3 条评论
博主 opteroncx
有什么问题可以来留言
博主 akizya
你好,我发现我的libcaffe和caffe项目生成都没有问题,但是别的项目都存在error lnk2001无法识别外部符号的问题
博主 opteroncx
@akizya 你是下的是bvlc的caffe么?有没有完整的错误日志发一下