本篇整理一下caffe安装的步骤。装caffe已经很长时间了,可能细节上有遗漏的地方,望指正。
一、下载caffe
登录网址:https://github.com/BVLC/caffe/tree/windows
(1)使用git clone 拷贝到本地。
(2)直接点击右侧按钮下载到本地并解压
二、安装CuDNN、CUDA(可选)
如果想使用GPU的就要下载,如果只想用CPU那就可以忽略。
1. CUDA下载地址:https://developer.nvidia.com/cuda-downloads
2. CuDNN下载地址:https://developer.nvidia.com/accelerated-computing-toolkit
3. 将CuDNN的安装包解压到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 目录下。
三、安装caffe
1. 没有安装CUDA的情况下,默认是CPU_ONLY的。如果安装CUDA之后还是想要安装CPU版本可以到caffe-windows目录下的CMakeList.txt文件下更改。
2. 安装Ninja,Visual Studio。
3. 用命令行切换到caffe-windows/scripts目录下,执行build_win.cmd。(建议开着VPN进行安装)
(CMakeList.txt的部分截图)
更多细节请参考https://github.com/BVLC/caffe/tree/windows
(真的很佩服能坚持写博客的大神们,真不是一个简单的事情。)