虽然网上的教程很多,但是安装的时候总会遇到不同的问题。因此记录一下自己的安装过程,希望能帮助到更多的人。
首先是必要的软件和工具:
1、VS2015 我用的是community版
2、Python3.5 https://www.python.org/
3、Cmake(安装3.4以上版本) https://cmake.org/download/
4、cuda8.0(目前8.0只能是跟VS2015配合安装) https://developer.nvidia.com/cuda-80-ga2-download-archive
5、cuDNN v5(注意版本,好像是v6以上会有错误) 需要注册NVIDIA的账号下载https://developer.nvidia.com/cudnn
6、caffe官方的包(用git或者下载zip文件都可以) https://github.com/BVLC/caffe/tree/windows
7、Ninja https://ninja-build.org/ (下载好后将exe文件放入cmake的bin文件夹)
以上都是用的64位
安装完Python和Cmake以后要将bin所在的文件夹添加到系统环境变量,具体操作如下:
这两个路径添加到我的电脑-属性-高级系统设置-高级-环境变量-系统变量-双击path-加;添加路径
接下来安装好CUDA8.0 然后把cuDNN下载好的压缩包解压,将三个文件夹bin lib 和include放到CUDA的文件夹内
这里project是准备放caffe的文件夹,可以自行调整路径。如果不是下载zip文件的就用第一行代码下载caffe安装包。用zip的直接解压到指定文件夹就行了。
在执行C:\Projects\caffe> scripts\build_win.cmd 之前,可以用notepad++修改一下参数。
用notepad打开 scripts\build_win.cmd 文件, 第九行参数改为0,使用CUDA
if NOT DEFINED CPU_ONLY set CPU_ONLY=0
然后在cmd中输入代码
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
开始安装(漫长的过程 我用了20分钟左右)(中间一个包的下载出了点问题,重试了几次)
编译完成后,在caffe的build文件夹里我们可以看到编译出来的文件。用VS2015打开caffe.sln,开始编译
编译完成就安装完成了