win10下caffe安装教程

安装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版本:

下载完成后进行安装,安装过程采用默认地址(不推荐修改默认安装地址):

https://img-blog.csdn.net/20170908104655390?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvdTAxMzE2NTkyMQ==/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast

https://img-blog.csdn.net/20170908104835358?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvdTAxMzE2NTkyMQ==/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast

https://img-blog.csdn.net/20170908105110845?watermark/2/text/aHR0cDovL2Jsb2cuY3Nkbi5uZXQvdTAxMzE2NTkyMQ==/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70/gravity/SouthEast

点击同意继续下一步,等待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)

 

 

 

猜你喜欢

转载自blog.csdn.net/u011947630/article/details/81346273