Caffe+cuda7.5+cuDNN v5.1 在windows10下安装

来源:互联网 发布:linux vim 保存并退出 编辑:程序博客网 时间:2024/06/05 05:01

一般来说caffe很少在windows下安装,能在ubuntu就在ubuntu,但是之前安装双系统瞎折腾导致两个系统同时不能用,搞了一天重装系统后现在只能使用windows硬装。
目前看到的windows的安装方案分为两种,第一种是伯克利官方伯克利官方的安装方案,另外一种是 微软的安装方案(目前停止维护)当然还有快糙好猛的happynear方案 (微软的方案傻瓜版)。看介绍还是happynear方案更简单,不过这里选择的是微软的方案。

要求

先决条件:
Visual Studio 2013 (社区版)
Anaconda2-2.4.1-Windows-x86_64
CUDA Toolkit 7.5
cuDNN v5.1(亲测5.0也可以)

步骤

在清华大学的开源软件库找到并安装anaconda,anaconda会顺便安装好python2.7,需要将其设置为python默认路径。随便测试一下可以用就行。

安装cuda7.5, 现在官网没有下载,只能找到第三方包,对应自己的系统版本。可能会在安装过程中出现与当前驱动不兼容的问题,如果出现安装失败的问题,可能会导致vs和cuda不能正常关联,可以选择自定义模式仅安装cuda相关toolkit和 Visual Studio关联的驱动。将下载好的cuDNN解压缩,会有一个cuda文件夹,在环境变量找到cuda7.5的路径,将压缩包里的cuda中三个文件夹中的文件放入对应的cuda7.5文件夹中的相应地址。

这里写图片描述

这里写图片描述
在cmd中测试cuda,输入

nvcc -V

显示

这里写图片描述

并且vs中新建一个project看到Nvidia cuda的项目。

这里写图片描述

这样才算cuda装好了!简直全是坑。


虽然anaconda足够强大,安装了几乎所有有用的库,但是由于接下来要安装的nvidia的可视化组件digits要求迷之复杂,所以全部依着他的要求进行安装。以下是必须用到的库。requirements.txt

Pillow>=2.3.0,<=3.1.2
numpy>=1.8.1,<=1.11.0
scipy>=0.13.3,<=0.17.0
protobuf>=2.5.0,<=2.6.1
six>=1.5.2,<=1.10.0
requests>=2.2.1,<=2.9.1
gevent>=1.0,<=1.1.0
gevent-websocket==0.9.3
Flask==0.10.1
Flask-WTF>=0.11,<=0.12
wtforms>=2.0,<=2.1
Flask-SocketIO==2.6
setuptools>=3.3,<=20.7.0
lmdb==0.87
h5py>=2.2.1,<=2.6.0
pydot>=1.0.28,<=1.0.29
psutil>=1.2.1,<=3.4.2
matplotlib>=1.3.1,<=1.5.1
scikit-fmm>=0.0.9

numpy必须是+mkl版本的,pillow不能使用3.0以上的,这里下的是2.6.1,而且GraphViz在windows的安装也很麻烦需要配置path。
requirements.txt可以在digits找到,在cmd执行

pip install -r requirements.txt

就很帅,之前一个个对比一个个下载真是要骂娘。但是有几个包可能不能用pip安装,实在没办法可以到网上自己找着下。总之解决了就行。
测试翼搏”
这里写图片描述

准备工作完成。
使用drawnet测试一下,C:\caffe-master\Build\x64\Release\pycaffe>python draw_net.py lenet.prototxt net.bmp --rankdir=BT
1
这里写图片描述
成功


将windows caffe解压到任意路径(C:\caffe-master),复制.\windows\CommonSettings.props.example 到.\windows\CommonSettings.props

按照需求改什么cuDnn,python,matlab,python接口这些东西。由于和默认参数没有什么变化所以这里直接就是只要改一下

<PythonDir>C:\Anaconda2</PythonDir> <PythonSupport>true</PythonSupport>

用vs2013打开./windows\Caffe.sln,不能用vs2015。设置成release模式,将project选项中的将警告理解为错误改成no,

这里写图片描述

不然会出现错误

LNK1181: cannot open input file ‘libcaffe.lib’ C:\caffe-master\windows\test_all\LINK test_all

然后就是到build里面去编译,需要的组件会自己下载,半个小时之后差不多就完成了。
这里写图片描述

折腾了两天,基本上网上的坑都踩过了,有的教程自己就是坑,还是看原版的指南安装完全按部就班比较好。

0 0
原创粉丝点击