参考链接:https://www.cnblogs.com/leoking01/p/6951750.html
基本环境:win10 GPU 1050TI CUDA8.0 cuDNN6.0
vs2015 cmake3.10
上述环境资料比较多就不赘述了,下面主要介绍源码编译vs2015工程
1、caffe源码下载
源码地址链接:github:https://github.com/BVLC/caffe/
我是采用方式下载:
新建caffe文件夹,dos窗口进入文件夹下
cd E:\caffe\caffesrc
git clone https://github.com/BVLC/caffe.git
cd caffe
git branch -a
git checkout windows
2、利用cmake-gui生成vs2015工程文件
设置源码路径:E:/caffe/caffesrc/caffe
设置build生成路径:E:/caffe/caffesrc/caffe/Cmake_build/x64
点击configure,选择vs版本,Visual Sutdio 14 2015 Win64
tooset 默认值,空
勾选[Use default native compilers]
需要下载依赖,网速比较好的话,可以直接点击configure,进行生成,如果网速不好,可以去网站下载
我的网速不好,我去官网下载,具体网址:打开E:\caffe\caffesrc\caffe\cmake文件夹,找到WindowsDownloadPrebuiltDependencies.cmake这个文件
用Notepad++打开,查看
我选择的python3.5,vs2015所以网址是 https://github.com/willyd/caffe-builder/releases/download/v1.1.0/libraries_v140_x64_py35_1.1.0.tar.bz2
进行下载
下载完成后需要将文件解压,并且需要copy到C:\Users\user\.caffe\dependencies\libraries_v140_x64_py35_1.1.0
再次执行configure 会出现如下BLAS配置错误。
把BLAS属性设置为OPEN
设置python_version属性为3(或者2) #3表示调用3.5,2表示调用2.7
再次configure,会出现Boost_PYTHON-PY*_LIBARY_*选项warning,可忽略,后面会自动找到Boost的正确版本和路径
configure成功后,点击generate生成vs工程成功
3、编译vs2015工程
点击open project打开工程
选择release版本
右击ALL_BUILD,点击生成,开始编译
生成成功38项
dos窗口进入生成目录E:\caffe\caffesrc\caffe\Cmake_build\x64\tools\Release
输入caffe.exe -version
出现上面画面说明编译成功
再编译debug的时候会出现一个坑
有10多个生成错误,显示找不到python35.lib
需要将上述错误选项的链接输入项中的D:\Program Files\Python35\libs\python35_d.lib改为D:\Program Files\Python35\libs\python35.lib
重新编译即可通过