开发环境Win10_X64 VS2015
注意看caffe的官网介绍,对各个库都有版本要求的。官网永远是最靠谱的资料。
一、安装CUDA
cuda_8.0.61_win10.exe下载地址:
安装后有如下环境变量(自动生成,不用你自建):
CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
CUDA_PATH_V8_0=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
NVCUDASAMPLES8_0_ROOT=C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0
NVCUDASAMPLES_ROOT=C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0
NVTOOLSEXT_PATH=C:\Program Files\NVIDIA Corporation\NvToolsExt\
二、安装CUDNN
谈不上安装,其实就是下载现成的库到你的电脑。
cuDNN下载需要注册个账号。下载cudnn-8.0-windows10-x64-v5.1-zip
https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v5.1/prod_20161129/8.0/cudnn-8.0-windows10-x64-v5.1-zip
https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v5.1/prod_20161129/8.0/cudnn-8.0-windows10-x64-v5.1-zip
解压后是这个样子
官网介绍了两种配置方法,二选一哟!
方法一,直接拷贝到cuda安装目录下对应的同名文件夹(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)
方法二,
专门为
CUDNN
建个位置(例如
C:\Program Files\cuda
),同时要添加到环境变量:
CUDNN_ROOT=C:\Program Files\cuda
我推荐方法一。
三、安装anaconda2,安装完后,python库路径会自动添加到环境变量。
anaconda2对应的python版本是2.7,也有文章说安装3.5,但是我没试过。
四、编译caffe
首先选对源码,既然是在windows平台下,就选windows分支:https://github.com/BVLC/caffe/tree/windows
直接下载下来然后解压到文件夹D:\Workspace2015\caffe_c++\,同时在该文件夹下创建build空文件夹。
如果提示
CMake Error at C:/Program Files/CMake/share/cmake-3.10/Modules/FindPackageHandleStandardArgs.cmake:137 (message):Could NOT find Atlas (missing: Atlas_CLAPACK_INCLUDE_DIR
那么将下图Atlas改为Open
在github主页下方有详细的要求和步骤。