win10+VS2015+cuda+cudnn安装与配置Caffe

需要使用的工具:

  • Git: 可直接使用GitHub提供的工具GitHubDesktop
  • Visual Studio 2015: 可以使用免费的社区版,安装的时候一定要勾选开发组件,确保C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin\amd64下有cl.exe程序,否则没有安装完全,后面会出错。
  • Python: 下载64位版本,建议使用Anaconda3,新建的环境使用python3.5,然后将python路径加入系统环境变量path中
  • CUDA 8.0
  • cuDNN v5
  • CMake>3.4版本 不需要安装,但是要解压后将路径添加到系统路径下

我们一点一点分析:

  1. 首先必须安装好VS2013或者VS2015,这是因为Caffe的底层代码主要是C++。
  2. 必须安装好CMake,且版本要>=3.4。关于cmake的下载,cmake下载,最新的版本已经到了3.11了。已经下好的,可以打开cmd,试一试,如下:
cmake -version
  • 1

这里写图片描述

这里显示我的cmake版本为3.4,刚好满足最低需求。需要注意的是,这个cmake必须要已经添加在你的系统路径里面,因为在后面执行脚本的时候,要能直接使用,检验如下:

这里写图片描述

如果你想在python中使用Caffe,那么就需要安装好python,(推荐使用Anaconda,至于版本,只支持2.7和3.5的,可以在官网上下载对应的,Anaconda下载),需要注意的是python也需要添加到环境变量中,检查如下:

这里写图片描述

这里写图片描述

如果需要在matlab中使用Caffe,就要装好matlab。

所以确保了这些,就可以继续往下走了:

这里写图片描述

按照下面的代码执行(根据自己的文件目录,不一定按照它的来):

C:\Projects> git clone https://github.com/BVLC/caffe.git
C:\Projects> cd caffe
C:\Projects\caffe> git checkout windows
C:\Projects\caffe> scripts\build_win.cmd
  • 1
  • 2
  • 3
  • 4

一般执行到前面3步都没有问题。文件结构大概是这样的:

这里写图片描述

我的Caffe路径在E:\DeepLearning\caffe这个时候里面还没有build文件夹,这是后面cmake生成的。

接着打开scripts\build_win.cmd 这个文件(比如用Notepad++),更改里面的一些配置(由于没有用到APPVEYOR,所以直接看else后面的):

这里写图片描述

需要改动的主要如下: 
1. 根据自己的VS版本,改变VS版本号,VS2015版本号就是默认值14 
2. 把Ninja设为0,因为这里不需要用到它 
3. 把CPU_ONLY设置为0,因为这里使用到GPU 
4. 根据自己的python版本,改变版本号

改完后再执行scripts\build_win.cmd,这时我遇到了这样的问题:Could NOT find PythonInterp (missing: PYTHON_EXECUTABLE) (minimum required is "3.5") ,我明明配置好了python路径,而且也可以直接在cmd里面运行。搜索了很久,还是没有解决,后来突然看到了这段代码:

这里写图片描述

虽然并没有运行,但是它手动写出了python的路径,所以我把它也放到了else那段里面了,如下:

这里写图片描述

然后就解决了这个问题,这里需要注意,如果这里的python路径没有加入系统路径下,这里也是不能成功的。后来再执行scripts\build_win.cmd,发现还是报错,一个名为libraries_v140_x64_py35_1.1.0.tar.bz2的依赖包无法下载成功。于是打开...\caffe\cmake\WindowsDownloadPrebuiltDependencies.cmake文件,看看它的下载地址(后面需要加上具体的路径,比如我的是(https://github.com/willyd/caffe-builder/releases/download/v1.1.0/libraries_v140_x64_py35_1.1.0.tar.bz2):

这里写图片描述

下载地址就是这个。于是手动去这个地址上把包下载下来,当时也是下了很多次,一开始网速都很慢(10KB/s的样子。。) 
把下载好的文件放在 ...(你的c盘当前用户目录)\.caffe\dependencies\download 这个目录。然后打开...\caffe\cmake\WindowsDownloadPrebuiltDependencies.cmake这个文件,把下载部分注释掉(红色圈子里面):

这里写图片描述

并且打开...\caffe\scripts\download_prebuilt_dependencies.py文件,注释掉下载部分:

这里写图片描述

接着再运行build_win.cmd,就可以顺利进行了,它会把刚刚下载的那个压缩包解压,然后慢慢编译。。。

这里写图片描述

大概用来7分多钟,就完成了:

这里写图片描述

看到这一幕的我开心地笑了。。。折腾了一天,终于配置得差不多了。于是又开始配置python接口:

这里写图片描述

这里的坑倒不是很多,因为安装了Anaconda,安装包特别方便(当然还是要看网速的心情了),但是有些是不用装的,比如cmake,ninja,graphviz,按照上面的图来就ok了。最后就可以在python的环境中导入caffe了!

这里参考了http://blog.csdn.net/qxconverse/article/details/76376718


猜你喜欢

转载自blog.csdn.net/u010532666/article/details/79382606