Win10 VS2015 caffe CUDA8.0 GTX1070安装指南

来源:互联网 发布:java date类 编辑:程序博客网 时间:2024/06/05 15:58

Win10 VS2015 caffe CUDA8.0 GTX1070安装指南

  • caffe vs2015版本代码下载
    下载地址:[https://github.com/happynear/caffe-windows]
    下载3rdparty: caffe依赖的3方库,已经用vs2015 x64编译
    编译前的准备
    将 .\windows\CommonSettings.props.example 拷贝一份,并改为 .\windows\CommonSettings.props

  • CUDA 安装
    cuda 下载地址:下载需要的版本,我这里下载的v8.0

    如果你没有GPU, 就跳过安装CUDA的步骤,并在 CUDA .\windows\CommonSettings.props 中CPU_ONLY 设置为true, 否则老老实实安装。如果只用CUDA、不需要CuDNN ,则UseCuDNN 设置为false

  • CuDNN安装
    CuDnn下载地址:下载需要的版本,我这里下载的V5.1
    这里有两种安装方法
    一:
    下载后直接解压,将lib,include, bin文件中的东西分别拷贝到CUDA安装目标里面相应的lib,include,bin 文件夹中。
    二:
    下载后直接解压,然后将cudnn的路径写到 .\windows\CommonSettings.props中

-如果你使用的是c++接口,就可以直接编译libcaffe项目,生成caffe的lib文件。

  • Python 接口
    1、首先下载python,并安装,这个网上很多例子
    2、用pip安装 numpy scipy scikit-image protobuf 等库
    3、安装好后,注意要讲python的安装路径设置到 .\windows\CommonSettings.props 就可以开始编译pycaffe项目,生成caffe 的caffe 接口库_caffe.pyd。
    4、将这个caffe文件夹复制到D:\Anaconda2\Lib\site-packages中,然后尝试使用import caffe。import可能会出现typeerror:init()got an unexpected keyword argument ‘syntax’这样的错误,解决的办法是在D:\Anaconda2\Lib\site-packages\caffe\proto中选择caffe_pb2.py文件,将文件中所有含有syntax的语句注释掉即可。

需要看更多的机器学习资料,请点击这里

联系方式:QQ472338311
github: https://github.com/guozhongluo

0 0
原创粉丝点击