1. 下载cuda8.0 cudnn5 anaconda
https://developer.nvidia.com/cuda-80-ga2-download-archive
https://developer.nvidia.com/cudnn
https://www.anaconda.com/download/#windows(安装anaconda3)
安装anaconda,cuda,cudnn
cudnn下载好后,把include和lib目录复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0(根据你的安装目录)
使用anaconda 安装python3.5(因为caffe需要安装python3.5,必须是3.5版本,切记!!)
conda install python=3.5
进入windows cmd,运行python -V 如果显示版本是3.5,那就OK啦~
2. 下载caffe源码,以及windows需要的依赖
源码:https://github.com/BVLC/caffe/tree/windows
3. 修改 caffe-windows\scripts\build_win.cmd 为如下:
if NOT DEFINED WITH_NINJA set WITH_NINJA=1
if NOT DEFINED CPU_ONLY set CPU_ONLY=0
if NOT DEFINED PYTHON_VERSION set PYTHON_VERSION=3.5
-DUSE_NCCL:BOOL=!USE_NCCL! ^
-DCUDA_ARCH_NAME:STRING=%CUDA_ARCH_NAME% ^
-DCUDNN_ROOT=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA ^ (这个是第172行加入的东西)
"%~dp0\.."
3.下载cmake gui,打开cmake ,设置源码路径,和生成路径(自己设置),点击configure 它会自动帮你下载好依赖
如果 cmake 出现错误
Could NOT find Atlas (missing: Atlas_CLAPACK_INCLUDE_DIR
Atlas_CBLAS_LIBRARY Atlas_BLAS_LIBRARY Atlas_LAPACK_LIBRARY)
解决办法 : cmake中BLAS值改为OPEN
点击generate 完成!
注:如果下载依赖出错,你可以手动点击这个链接下载 https://github.com/willyd/caffe-builder/releases/download/v1.1.0/libraries_v140_x64_py35_1.1.0.tar.bz2 ,然后放到dependency目录
4. 通过VS2015 打开caffe.sln
debug 和 release版本都生成一下,然后右键点击INSTALL的那个项目,点击“仅用于项目” , 再点击生成,这样,我们就装好啦~~
补充:本人使用python 2.7版本也是可以的。