本教程是caffe在windows系统上的配置方法,而且是github上官方BVLC/caffe目前推荐的配置方法,并不是使用微软的caffe-windows。
前期准备
- 首先从BVLC/caffe上下载caffe源码,选择好路径后下载。下载过后,会得到一个caffe-windows.zip文件。
- 将该文件解压到你指定的位置,进行下一步。
- 确定你是不是有NVIDIA的GPU,需不需要用GPU加速,如果不用GPU加速的话,可以跳过下面的cuda安装和cudnn配置,直接配置caffe;如果需要用GPU,则需要在NVIDIA官网上分别下载安装cuda和cdunn,下载地址在下面给出。
- 按照官方建议,需要使用cmake,所以需要先下载 cmake并安装,给出的这个cmake链接是64位的,直接下载安装,全英文界面,所有选项保持默认即可,一直点击下一步,直到安装完成。
环境配置
cuda和cudnn配置(不是使用NVIDIA GPU加速的话请掠过)
cuda和cudnn都是NVIDIA(英伟达)公司提供给开发者使用的运算平台,简而言之就是用GPU干你想干的事。
- 下载并安装cuda,一定要选择对应的版本,要不然会安装失败的。现在我演示的是64位win7,所以就选择对应的版本下载。安装过程比较简单,双击安装,选项建议都默认(有些人不喜欢把东西放在C盘,建议还是放在C盘,毕竟是开发工具),一直点击确认或者下一步,直至安装成功即可。
- 下载cudnn,这是用来加速深度网络训练的。下载cudnn需要登陆,你可以直接用邮箱注册一个,然后登陆下载就好了。
- 下载完后,你会发现,cudnn不是一个软件,而是一个压缩包,我下载的就是:cudnn-8.0-windows10-x64-v6.0.zip(有人私信说caffe还是没有编译成功,我看了一下发现是cudnn版本问题,前面的cuda安装的是8.0,所以cudnn应该用的是cudnn-8.0-windows10-x64-v6.0.zip,而不是其他版本)。这个包需要解压缩,得到一个名为cuda的文件夹,为了不与前面安装的cuda混肴,用红色表示。
- 配置cudnn。cudnn解压后得到的cuda文件夹下有bin、include、lib三个子文件夹,现在需要将这三个文件夹下的东西分别放到之前cuda的安装目录下的对应文件夹下。之前已经安装过cuda,cuda的安装目录默认为:C:\Program Files\NVIDIA GPU Computing Toolkit\,现在就只需要:
-将cudnn解压后得到的cuda文件夹下的bin下的cudnn64_6.dll(cudnn后面的数字可能不一样,这就是版本问题,如果是cuda 8.0 建议搭配我这个版本的cudnn,否则可能会出现问题)文件拷贝到cuda安装目录下的bin目录下;
-将cudnn解压后得到的cuda文件夹下的lib文件夹下的x64文件夹下的cudnn.lib拷贝到cuda安装目录下的lib文件夹下的x64目录下
-将cudnn解压后得到的cuda文件夹下的include文件夹下的cudnn.h拷贝到cuda安装目录下include文件夹下
可能上面讲的可能有点啰嗦,但是希望你们能看懂啦。