从零开始学caffe(八):Caffe在Windows环境下GPU版本的安装

之前我们已经安装过caffe的CPU版本,但是在MNIST手写数字识别中,我们发现caffe的CPU版本运行速度较慢,训练效率不高。因此,在这里我们安装了caffe的GPU版本,并使用GPU版本的caffe同样对手写MNIST数字集进行训练。

step1: 安装CUDA

首先需要准备好NVIDIA的显卡,然后下载安装CUDA,在这里我安装的是CUDA8.0下载地址。CUDA的安装过程比较简单,在这里不再赘述(但有些时候会出现CUDA无法安装的问题,可能是版本不对应的问题,这时候要选择和自己电脑版本相对于的CUDA进行安装)。
在这里插入图片描述
在完成CUDA的安装后,注意要把CUDA安装目录下的bin和lib\x64添加到Path环境变量中
在这里插入图片描述

step2:cuDNN下载

在安装完CUDA后,下载与CUDA版本对应的cuDNN进行安装,下载地址
在这里插入图片描述

在完成cuDNN下载后,解压压缩包,把压缩包中bin,include,lib中的文件分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0目录下对应目录中,完成cuDNN的安装。

step3:修改CommonSettings.props文件

在完成CUDA和cuDNN的安装后,因为之前CommonSettings.props文件中的配置的是CPU版本,因此我们需要对此进行必要的修改。

<CpuOnlyBuild>false</CpuOnlyBuild>
<UseCuDNN>true</UseCuDNN>
<CudaVersion>8.0</CudaVersion>
<PythonSupport>false</PythonSupport>

step4:编译

用VS2013打开Caffe.sln,修改属性后重新生成解决方案。
在这里插入图片描述
然后将E:\caffe-windows\python下的caffe文件夹复制到C:\Users\MyPC\Anaconda3\envs\py2\Lib\site-packages下。至此,GPU版本的caffe已经配置完毕。

step5:测试

我们可以用刚刚配置完的GPU版本caffe对之前的MNIST手写数字集进行识别,将lenet_solver.prototxt文件中的CPU模式改成GPU,然后运行train.bat
在这里插入图片描述
整个GPU训练时间在1分钟左右,和之前CPU训练的速度相比加快了很多(debug版本训练耗时一个半小时,release版本消耗十多分钟)

猜你喜欢

转载自blog.csdn.net/dagongsmallguy/article/details/83352503
今日推荐