Win10+VS2013+CUDA7.5 Caffe 配置过程

本文详细介绍了在Windows 10系统中,如何配置CUDA 7.5、VS2013以及Python环境来搭建Caffe。首先,需要安装Python 2.7并添加相关路径到系统变量,然后通过.whl文件安装numpy、scipy等库。接着,安装scikit-learn和MATLAB。在安装Caffe之前,需按照GitHub上的指南进行预处理。遇到CUDA任务加载错误时,需安装.NET Framework 3.5。最后,下载MNIST数据集,运行run_mnist.bat进行训练,确保Caffe正常工作。此外,还介绍了如何处理protobuf的版本冲突,确保Python接口的正常使用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

  1. 准备工作

    1. Python:https://www.python.org/downloads/,建议下载 2.7.X 版本,工具全面兼容。安装后:
      • 将 Python 的根目录加载到系统变量中:
      • 同样地,将 Python 根目录中的 Python27\Scripts 文件夹同时加入系统变量 path 中(为了在命令行中方便使用 pip 命令)
    2. 原生的 Python 缺少机器学习必须的计算库(numpy, scipy)、绘图库(matplotlib)、功能库(boost.python)等等,现在 Python 推荐相关库的安装是使用 .whl 压缩文件。推荐一个非常好的第三方网址,总结了所有常用的库,而且维护及时:
      https://www.lfd.uci.edu/~gohlke/pythonlibs/
    3. 安装 .whl 压缩文件的方法是:使用命令行(cmd)进入 .whl 文件所在的文件夹,输入命令:

      pip install [.whl 文件名]

      安装 numpy, scipy 库之后,命令行中输入以下命令:

      python
      import numpy
      import scipy

      只要不会报错即说明尚未安装成功

    4. scikit-learn 是基于 Python 中 numpy, scipy, matplotlib 库的机器学习库,官网地址:http://scikit-learn.org/stable/index.html;Python 官网的下载地址:https://pypi.python.org/pypi/scikit-learn,.exe 文件安装;简易教程:http://kukuruku.co/hub/python/introduction-to-machine-learning-with-python-andscikit-learn

    5. MATLAB下载安装:随便搜索一个天朝绿色版装上就好……你懂的
  2. 正式安装

    1. 安装之前需要完成上面的 Python 过程,然后请完成此页面(https://github.com/Microsoft/caffe)的【Build】之前部分的安装步骤,从而才能使得 caffe 神经网络工具可以利用 GPU(CUDA),并且在 Python、MATLAB 中顺利使用

    2. 之后,下载此篇博客(http://blog.youkuaiyun.com/happynear/article/details/45372231)的文件、编译文件。编译 MainBuilder.sln 时可能出现的问题:

  3. 测试 caffe

    1. http://pan.baidu.com/s/1mgl9ndu 下载已经转换好的MNIST的leveldb数据文件,解压至./examples/mnist文件夹中
    2. 利用 windows 命令行(CMD)进入根目录,直接调用该目录下的 run_mnist.bat 即可开始训练
    3. 训练日志会保存在./log文件夹中,以 INFO 开头,txt格式的日志文件中
  4. Python 接口使用 caffe

    1. 首先我们回到步骤2:正式安装中的所下载的文件夹:\caffe-windows-master\python,创建新的系统环境变量 PYTHONPATH 存储此位置:
      这里写图片描述
    2. 安装谷歌的开源工具:protobuf 并且版本应该是 3.X,但是这里会有一个问题,如果已经重复安装了 protobuf,并不会覆盖原有的版本,而是两个版本共存从而造成问题。因此我们要做的第一件事是卸载原有版本的 protobuf,命令如下(注意:调用以下命令不能一次性卸载之前安装的所有版本的 protobuf,只会删除最近一次安装的 protobuf;为了干净卸载,建议多次调用以上命令,直至出现:Cannot uninstall requirement protobuf, not installed 提示):

      pip uninstall protobuf

    3. 重新安装 protobuf:

      • 下载 https://github.com/google/protobuf/releases 页面中的 protoc-$VERSION-win32.zip 文件,将解压之后的 protoc.exe 放入 C:\Windows\System32 中
      • 下载 https://github.com/google/protobuf 页面中所有源码,之后在命令行中进入 protobuf-master\python 文件夹,依次输入三个命令:

        python setup.py build
        python setup.py test
        python setup.py install

      • 三条命令都没有报错即说明最新版本的 protobuf 安装成功

    4. 测试 caffe 能够在 Python 中使用。命令行输入 python 进入交互编辑界面后输入:import caffe 不报错即证明可以顺利使用
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值