1.   打开examples\mnist目录下的lenet_solver.prototxt文件,可以看到solver_mode: GPU这一选项,可以设置为GPU或者CPU

net和snapshot_prefix  设置为绝对路径

windows+caffe+mnist识别

2.   打开examples\mnist目录下的lenet_train_test.prototxt文件,修改第一个和第二个layer的source为第一步生成的训练测试数据,

backend 需要与source   mnist_train_leveldb对应,有的是lmdb如下图所示:  source最好是绝对路径(减少不必要的麻烦)
windows+caffe+mnist识别

3.打开cmd命令行,输入

windows+caffe+mnist识别

便可以进行训练

4.   开始训练,如果出现下面错误

那么说明当前设置的CUDA的architecture与电脑的CUDA不匹配

打开CommonSettings.props可以看到CudaArchitecture选项,设置成电脑GPU的Architecture,重新编译之前的caffe工程。

说明:编译之前,一定要确定对应的CudaArchitecture是正确的,查看方法:右键libCaffe工程->属性

windows+caffe+mnist识别

点开CodeGeneration对话框,下面有一个宏按钮,点击如下,找到CudaArchitecture,查看值是否正确(我的电脑GPU计算能力是3.0,因此设置为compute_30,sm_30)

windows+caffe+mnist识别

4 测试

windows+caffe+mnist识别

得到测试结果

 

.

相关文章:

  • 2021-06-23
  • 2021-11-29
  • 2022-12-23
  • 2021-11-12
  • 2022-02-27
  • 2022-12-23
  • 2021-11-29
  • 2022-01-22
猜你喜欢
  • 2021-12-19
  • 2021-05-20
  • 2021-05-18
  • 2022-12-23
  • 2021-10-01
  • 2021-07-09
  • 2021-11-26
相关资源
相似解决方案