win10+vs2013+cuda8.0+caffe

本文详细介绍了在Windows 10环境下使用VS2013 Community、CUDA 8.0、Caffe和cuDNN搭建深度学习环境的过程,并通过测试MNIST数据集验证配置的有效性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1,首先说下环境和配置

配置:

环境:windows10,vs2013 community,cuda8.0,caffe,cudnn4
注意:先要安装好显卡驱动(我的显卡是1070),这里的例子只开通了matlab接口,需要用python可以安装anacoda并打开python接口。

2,准备文件

例子中用的文件:链接:http://pan.baidu.com/s/1hsBNI3i 密码:cynd
注意一:处理cudnn有两种方式,其一是在vs的属性文件中给出目录;另一是将cudnn的文件放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0的相应目录中;这里我用的是第二种方式。
注意二:解压出来的caffe-master应该有两层目录名字都是caffe-master,第一次编译的时候我直接去掉了一层,(因为我的小小强迫症),但是发现其实程序会在第一层caffe-master的下面新建一个NugetPackages文件夹来放置第三方库,所以实际上这里不要去掉其中一层caffe-master目录。
注意三:先编译libcaffe,再编译caffe,最后需要什么工程就编译哪个,编译过程中的错误会之后找个时间写一篇专门的博客。

3,测试mnist

需要下载的四个训练和测试数据已经在前面的分享文件夹中,下载即可;
转化数据格式到lmdb,用的是python代码,注意代码应该放在Build同目录下:

import os
import shutil
EXAMPLE='examples\mnist'
DATA='data\mnist'
BUILD='Build\x64\Release'
BACKEND='lmdb'
print "Createing "+BACKEND+"..."
path1=EXAMPLE+"\mnist_train_"+BACKEND
path2=EXAMPLE+"\mnist_test_"+BACKEND
if os.path.exists(path1):
shutil.rmtree(path1)
if os.path.exists(path2):
shutil.rmtree(path2)
s1=BUILD+"\convert_mnist_data.exe"
s2=DATA+"\train-images.idx3-ubyte"
s3=DATA+"\train-labels.idx1-ubyte"
s4=EXAMPLE+"\mnist_train_"+BACKEND
s5="--backend="+BACKEND
cmd=s1+" "+s2+" "+s3+" "+s4+" "+s5
print cmd
os.system(cmd)
t1=BUILD+"\convert_mnist_data.exe"
t2=DATA+"\t10k-images.idx3-ubyte"
t3=DATA+"\t10k-labels.idx1-ubyte"
t4=EXAMPLE+"\mnist_test_"+BACKEND
t5="--backend="+BACKEND
cmd=t1+" "+t2+" "+t3+" "+t4+" "+t5
print "cmd2="+cmd
os.system(cmd)

执行该代码后,会在examples\mnist下面出现两个文件夹,mnist_test_lmdb和mnist_train_lmdb;
写批处理文件来运行测试mnist:
Build\x64\Release\caffe.exe train --solver=examples\mnist\lenet_solver.prototxt
Pause
运行过程截图:

上图就能看到测试程序的结果,正确率等参数。祝成功,有问题留言!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

sophia_xw

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值