【caffe】VS2015下caffe配置(GPU版)——【caffe学习一】

1。环境准备

环境:

win10 x64

VS2015

cuda8.0 x64 (https://developer.nvidia.com/cuda-downloads)

cudnn5.1 x64(http://download.csdn.net/download/u012223520/9683325#comment)

python2.7 x64(https://www.python.org/downloads/release/python-2714/)

cmake3.10(https://cmake.org/download/)


其中cudnn5.1配置方法参照:

直接将下载到的cudnn5.1解压,将文件夹下bin,include,lib分别放入cuda目录,即 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

Use cuDNN

To use cuDNN the easiest way is to copy the content of the cuda folder into your CUDA toolkit installation directory. For example if you installed CUDA 8.0 and downloaded cudnn-8.0-windows10-x64-v5.1.zip you should copy the content of the cuda directory to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0


2。安装caffe

以上环境配置完成后,开始下载caffe

如果没有安装git(https://git-scm.com/download/win),需要先下载安装

安装完成后,在命令行下输入:

git clone https://github.com/BVLC/caffe.git
cd caffe
git checkout windows
:: Edit any of the options inside build_win.cmd to suit your needs
scripts\build_win.cmd


遇到错误,找不到Ninja,需要修改scripts/build_win.cmd文件,同时删除scripts/build/CMakeCache.txt(如果存在该文件的话)




修改完成后,运行build_win.cmd



如果成功找到cuda和cudnn会输出这样


否则输出


运行结束后,在scripts/build下找到Caffe.sln用VS2015打开,选择release x64,ALL_BUILD 生成(如果出现error MSB3073: 再次点ALL_BUILD重新生成),

最后在build/tools/Release目录中生成:




评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值