安装caffe前提要求:电脑系统win7,8,10都可以;显卡只有N卡可以用Cuda,利用GPU加速,非N卡就利用CPU就好;此外需要用matlab或python(推荐)编程的再下载对应软件。
1.安装cuda和cuDNN(无CPU则无需安装)
由于我所使用的编译器是VS2017,故只能安装CUDA 9.0以上版本。(CUDA8.0是与VS2017不兼容的,就是说如果想用CUDA8.0,必须使用VS2015以下的VS版本)
CUDA 9.1下载链接:https://developer.nvidia.com/cuda-downloads
按下图进行选择,其中Version根据你系统的版本进行选择,我的是Win10,故选择10版本:
下载完成后进行安装,安装过程采用默认地址(不推荐修改默认安装地址):
点击同意继续下一步,等待CUDA安装的结束,如果系统检查有误,有可能是没有安装VS2017,或者安装错了VS版本。
注:如果要卸载旧版本的cuda,可在控制面板中选择“程序和功能”,选择应用程序后右键卸载。
安装完成后,需要进行环境变量的设置:
在计算机上点右键,打开属性->高级系统设置->环境变量,如下图可以看到系统中多了CUDA_PATH和CUDA_PATH_V9_1两个环境变量。
还需要在环境变量中添加如下几个变量:
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.1
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
点击新建进行添加,拿第一个做例子如下图
设置完成之后,我们可以打开cmd来查看:
输入:set cuda
得到下图
验证cuda安装成功与否:
在cuda安装文件夹中有deviceQuery.exe 和 bandwidthTest.exe两个程序。重新打开cmd按下图输入输入 cd .. 两次,再输入cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\extras\demo_suite
分别输入deviceQuery.exe 和 bandwidthTest.exe,运行结果如下图。如果Rsult=PASS则说明通过,反之,Rsult=Fail 则需要重新安装。
如果以上都没问题,则说明CUDA9.0安装成功。
cuDNN下载链接:https://developer.nvidia.com/cudnn
注:官网下载cudnn时会要求注册,填完信息注册完成后,选着cuDNN v4 或者 cuDNN v5 (这取决于 CommonSettings.props 文件里标注支持的版本,建议选择 cuDNN v4 Library for Windows)安装尽量默认路径
2.下载caffe
下载链接如下:https://codeload.github.com/Microsoft/caffe/zip/master
解压,打开caffe-master文件夹,然后看到一个windows文件夹,然后继续打开windows文件夹,看到里面一个CommonSettings.props.example文件,复制出来一份,并改名字为CommonSettings.props,如下图所示:
然后修改如下配置:
<CudaVersion>9.1</CudaVersion> ----- cuda版本改为9.1(根据你安装的cuda版本修改)
<PythonSupport>true</PythonSupport> ----- 是否支持Python,我选择支持,所以改为true
<MatlabSupport>true</MatlabSupport> ----- 是否支持Matlab,我选择支持
<CuDnnPath>D:\cuda\cuda</CuDnnPath> ----- 因为我把cuDNN解压在了D盘,所以这里的路径需要根据你自己解压的实际路径修改
<PathonDir>D:\Anaconda3\</ PathonDir > ----- 因为我之前装了Anaconda3,所以我的路径是这样的,实际要根据你自己安装的pathon路径进行修改
<CudaArchitecture>compute_21,sm_21</CudaArchitecture>
如下图所示
3.编译
双击Windows下的Caffe.sln,检查解决方案中的项目,重点看libcaffe和test_all是否成功地被导入
第一次应该是这两个显示加载失败的,如上图所示,解决方法如下:
关闭VS,打开 cuda安装路径中的 MSbuildExtensions 文件夹(若是按默认路径安装的cuda,在C盘中搜索即可,如下图)
拷贝MSbuildExtensions 文件夹中的所有文件(如上图),复制到
C:/ProgramFile(x86)/MSBuild/Microsoft.Cpp/v4.0/V140/BuildCustomizations 文件夹下,替换目标中的文件(注:路径中的V140取决于你VS的版本,即在你电脑上不一定是V140)