总体步骤:

1. 安装显卡驱动

2. 安装CUDA8.0

3. 安装CuDNN_v5.1

4. 安装Caffe

5. Caffe example测试

 

详细步骤:

1. 安装显卡驱动

1.1 查看显卡型号。

    右键计算机->设备管理器->显示适配器

1.2 下载对应型号的最新显卡驱动。

    地址:http://www.nvidia.cn/Download/index.aspx?lang=cn

1.3 双击.exe文件安装

 

2. 安装CUDA8.0

2.1 登录官网下载CUDA8.0

    地址:https://developer.nvidia.com/cuda-toolkit

2.2 双击.exe文件安装

2.3 验证CUDA8.0已正确安装

2.3.1 打开cmd,输入nvcc -V。结果如下图。

    

 

2.3.2 编译CUDA8.0自带的samples

    在C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0目录下,用vs2013,打开Samples_vs2013.sln,在Release下,选中解决方案Samples_vs2013,重新生成解决方案。第一次编译的时候,会提示找不到”d3dx9.h”、”d3dx10.h”、”d3dx11.h”头文件,可从这里下载DXSDK_Jun10.exe,按照默认安装。(安装时可能出现错误S1023,可参考https://support.microsoft.com/en-us/help/2728613/-s1023-error-when-you-install-the-directx-sdk-june-2010解决)。重新打开Samples_vs2013,再次重新编译,全部编译成功。

 

2.3.3 测试CUDA8.0运行是否正常

    执行C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64\Release目录下的deviceQuery.exe和bandwidthTest.exe。结果如下图。

 

 

2.4 将CUDA添加到系统路径

    在计算机上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V8_0两个环境变量,接下来,还要在系统中添加以下几个环境变量: 
        CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0 
        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

    然后,在系统变量 PATH 的末尾添加: 
        ;%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%; 

    重新启动计算机。

 

3. 安装CuDNN_v5.1

3.1 登录官网下载CuDNN_v5.1

    地址:https://developer.nvidia.com/cudnn(需要注册账号)

 

3.2 下载完成后解压。

    cuda目录下有bin,include,lib三个文件夹。

 

3.3 拷贝目录至CUDA8.0的文件夹

    分别将cuda\bin、cuda\include、cuda\lib三个目录中的内容拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0对应的bin、include、lib目录下即可。

 

4. 安装Caffe

4.1 下载Caffe

    地址:https://github.com/Microsoft/caffe

 

4.2 用VS2013生成解决方案

    进入caffe-windows\windows文件夹,用VS2013打开caffe.lsn。然后,将libcaffe设为启动项,并改为Release模式,再点生成解决方案,生成NugetPackages文件夹。

    生成时遇到bug: error : CommonSettings.props not found!

    解决方法:将caffe-windows\windows下的CommonSettings.props.example复制一份,并重命名为CommonSettings.props

    生成时遇到bug: error MSB4019: The imported project "C:\Program Files (x86)\MSBuild\Microsoft.Cpp\v4.0\V120\BuildCustomizations\CUDA 7.5.props" was not found. Confirm that the path in the <Import> declaration is correct, and that the file exists on disk.

    解决方法:打开CommonSettings.props,将其中的<CudaVersion>7.5</CudaVersion>改为<CudaVersion>8.0</CudaVersion>

    生成时遇到bug: Error 8534 error C2220: warning treated as error - no 'object' file generated (..\..\src\caffe\util\math_functions.cpp) 

    解决方法:用VS2013打开alt_sstream_impl.hpp,点击保存。用VS2013打开math_functions.cpp,点击保存。

 

4.3 3rdparty

    在caffe源码的根目录下建立个3rdparty文件夹,把文件解压进去就可以了。解压好之后,要将3rdparty/bin文件夹添加到环境变量的PATH中,这样才能让程序找到这些第三方库的dll。

    地址:http://pan.baidu.com/s/1bSzvKa

 

4.4 添加python接口

4.4.1 修改CommonSettings.props

    打开CommonSettings.props,将<PythonSupport>false</PythonSupport>改为<PythonSupport>true</PythonSupport>。

    将<PythonDir>C:\Miniconda2\</PythonDir>修改为<PythonDir>python安装路径</PythonDir>

 

4.4.2 安装anaconda

    地址:https://www.continuum.io/downloads/

    下载完成后,点击.exe文件安装。

 

4.4.3 安装必要的python包

    输入如下语句:

    conda install --yes numpy scipy matplotlib scikit-image pip

    pip install protobuf

 

4.4.4 修改pycaffe项目配置属性

    右键pycaffe,属性,C/C++附加包含目录,把python的<安装目录>\lib\site-package\numpy\core\include目录添加进去。

 

4.4.5 编译pycaffe

    编译pycaffe项目,编译后生成pycaffe。将<caffe_root>\Build\x64\Release\pycaffe\caffe文件夹复制到<python_root>\lib\site-packages。打开cmd,输入python,输入import caffe。若不报错则编译成功。

 

5. Caffe example测试

    这里以经典的mnist为例,测试刚刚编译好的windows版Caffe。

    打开cmd,cd进入<caffe_root>。输入Build\x64\Release\caffe.exe train -solver examples\mnist\lenet_solver.prototxt

    运行结果如下图:

    

    

    若网络正常训练,则表示Caffe安装成功~

 

参考链接:

http://blog.csdn.net/xzzppp/article/details/51510785

http://blog.csdn.net/lvfeiya/article/details/53325784

http://blog.csdn.net/d5224/article/details/51915290

 posted on 2017-02-21 15:53  黑芝麻12  阅读(7927)  评论(0编辑  收藏  举报