windows 10 环境安装caffe并测试mnist成功详解

本文主要总结windows 10 环境下编译安装caffe并成功测试DNN的详细过程:


软件包:

软件环境   visual studio 2013、windows_10_64bit

安装包      

caffe-master

https://github.com/Microsoft/caffe


cuda_8.0.44_win10.exe

https://developer.nvidia.com/cuda-downloads


cudnn-8.0-windows10-x64-v5.0-ga.zip

https://developer.nvidia.com/rdp/cudnn-download


Miniconda2-latest-Windows-x86_64.exe

https://www.continuum.io/downloads


Matlab2014a

http://cn.mathworks.com/index.html?s_tid=gn_logo


NVIDIA GeForce GTX960 driver

(驱动精灵自动安装)


硬件环境:

CPU  inter(R) Core(TM) i7-4790 CPU @ 3.60GHz

GPU  NVIDIA GeForce GTX960




(一)安装caffe

1、下载caffe-master到D:\

2、修改配置文件CommonSettings.props.example

   将D:\caffe-master\windows下的CommonSettings.props.example复制并重命名为 

   CommonSettings.props

   配置文件修改如下:



(二)编译caffe

1、导入caffe-master


双击Caffe.sln,即可将caffe-master导入visual studio 2013

2、编译caffe


当左下角显示“就绪”后,即可右击解决方案,选择“生成解决方案”,编译时会自动下载依赖包NugetPackages

共有16个项目,编译过程中可能会出现如下报错:

    a. 无法打开文件“python27_d.lib”

        解决方法:http://blog.csdn.net/junparadox/article/details/52704287

    b. "gpu/mxGPUArray.h" Not Found

        解决方法:http://cn.mathworks.com/matlabcentral/answers/141790-gpu-mxgpuarray-h-not-found

    c. 无法打开libcaffe.lib

        解决方法:直接对libcaffe重新生成,或者直接生成一次,当这个文件生成成功以后,重复一下上一步,对所有文         件再重新生成一下,这时候应该不会出现libcaffe.lib无法打开的错误了。

        其他一些bug,可以自行Google、百度一下都可以,注意一定要保证编译充分,编译过程要保证网络畅通。

3、Matlab\Python的安装详解Microsoft/Caffe/ReadMe

      网址:https://github.com/Microsoft/caffe

(三)测试mnist

1、下载mnist数据集

      train-images.idx3-ubyte、train-labels.idx1-ubyte、t10k-images.idx3-ubyte、t10k-labels.idx1-ubyte

      网址:http://yann.lecun.com/exdb/mnist/

      将训练集与测试集标签分别导入:

      D:\caffe-master\data\mnist\mnist_train_lmdb、D:\caffe-master\data\mnist\mnist_test_lmdb

2、制作lmdb格式数据集

      制作脚本create_mnist.bat(脚本路径 D:\caffe-master)

      源代码:

      .\Build\x64\Release\convert_mnist_data.exe .\data\mnist\mnist_train_lmdb\train-images.idx3-ubyte .\data\mnist\mnist_train_lmdb\train-labels.idx1-ubyte .\examples\mnist\mnist_train_lmdb   

echo.   
      .\Build\x64\Release\convert_mnist_data.exe .\data\mnist\mnist_test_lmdb\t10k-images.idx3-ubyte   .\data\mnist\mnist_test_lmdb\t10k-labels.idx1-ubyte .\examples\mnist\mnist_test_lmdb  
pause  

       双击“create_mnist.bat”生成相应lmdb格式mnist数据集:  

        \examples\mnist\mnist_train_lmdb

        \examples\mnist\mnist_test_lmdb 

2、编辑训练脚本

      制作脚本train_lenet.bat (脚本路径 D:\caffe-master

      源代码:

      SET GLOG_logtostderr=1
      "./Build/x64/Release/caffe.exe" train --solver=examples/mnist/lenet_solver.prototxt 
      pause

3、双击“train_lenet.bat”,测试mnist成功,界面如下:

      

      



  • 3
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值