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

之前我们已经安装过caffe的CPU版本,但是在MNIST手写数字识别中,我们发现caffe的CPU版本运行速度较慢,训练效率不高。因此,在这里我们安装了caffe的GPU版本,并使用GPU版本的caffe同样对手写MNIST数字集进行训练。 step1: 安装CUDA 首先需要准备好NVIDIA的显卡,然后下载安装CUDA,在这里我安装的是CUDA8.0下载地址。CUDA的安装过程比较简单,在这
相关文章
相关标签/搜索