caffe-windows配置

先检查python cudnn cuda版本号,不符合官方git要求的都是坑。

1.准备工作: 
第一步:下载Caffe-Windows(BLVC),地址:https://github.com/BVLC/caffe/tree/windows,并解压到指定的文件夹Caffe下。 
第二步:下载CuDNN5.1,并拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0对应的文件夹下。 
第三步:安装CMake3.7,并将其安装包下的bin路径到系统环境Path变量中。 
以上工作在安装Anaconda、VS、CUDA的前提。

2.修改配置文件:

打开Caffe文件夹下scripts的build_win.cmd脚本,使用Notepad++软件,修改以下内容: 
第一处修改: 
这里写图片描述 
第二处修改: 
这里写图片描述 
第三处修改: 
删除以下内容或者用“ :: ”注释掉。 
这里写图片描述 
完成以上三处操作并保存。

3.编译工作

第一步:在Caffe文件下shift+右键,点击“在此处打开命令窗口”,然后将Caffe\scripts下的build_win.bat文件拖到cmd窗口中,回车并编译。 
第二步:第一步会出错,因为下载libraries_v140_x64_py35_1.1.0.tar失败,这需要手动的下载。在 
下载地址:这里写链接内容,下载后将其拷贝到用户目录下.caffe\download目录下。 
第三步:下载好libraries_v140_x64_py35_压缩文件后,并拷贝该文件到第二步中的download文件加下,并重复第一步操作。

4.打开VS2015生成Release

用VS2015打开Caffe\build文件下的Caffe工程,VS2015开始加载,并无错误。

5.MNIST测试

第一步:下载MNIST的leveldb格式数据,解压两个文件夹mnist-train-leveldb, mnist-test-leveldb,修改成mnist_train_leveldb, mnist_test_leveldb,并拷贝到caffe\examples\mnist下。 
第二步:修改examples\mnist\lenet_train_test.prototxt,修改属性data param下的source和backend属性,如下图: 
第一个 
data_param { 
source: “绝对路径examples/mnist/mnist-train-leveldb” 
batch_size: 64 
backend: LEVELDB 

第二个 
data_param { 
source: “绝对路径examples/mnist/mnist-test-leveldb” 
batch_size: 100 
backend: LEVELDB 

第三步:在Caffe文件夹下创建一个run_mnist_leveldb.bat文件,内容如下: 
“./build/tools/Release/caffe.exe” train –solver=./examples/mnist/lenet_solver.prototxt 
pause 
(红色标记的极为重要,不可缺少) 
第四步:双击run_mnist_leveldb.bat即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值