【Caffe学习笔记】一 、环境安装 Caffe + cuda + windows10 + VS2015 安装笔记, win7也适用

时间:2021-04-03 16:43:38

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)

安装anaonda,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的那个项目,点击“仅用于项目” , 再点击生成,这样,我们就装好啦~~

【Caffe学习笔记】一 、环境安装     Caffe + cuda + windows10 + VS2015 安装笔记, win7也适用