背景
gpu的训练速度优于cpu,那么到底快多少呢?本博客拿caffe、cifar10做一下测试。
机器配置
系统:deepin 15.11
cpu:i7 7700
内存:32G 2400hz
显卡:gtx1080(中端)
mnist数据集
10000次迭代,使用lenet(lenet_solver.prototxt)。
gpu训练
耗费时间:14s
cpu训练
08:43:48
耗时 = 17:16:00 - 17:23:39 = 8分钟
cifar10数据集
迭代5000次,使用caffenet
gpu训练
耗时19s
cpu训练
08:49:57 - 09:16:19
耗时:27分钟
总结
- gpu大约是cpu的30-80倍
- 每次batch的size数据越大,gpu提升的速度更加明显