我可以在没有GPU的情况下训练深度卷积网络吗?

时间:2015-12-03 21:55:15

标签: deep-learning conv-neural-network

我正在考虑将卷积神经网络构建为跟踪系统应用程序。我感觉所有深层网络应用程序都需要使用GPU。是否有必要在像我这样的任务中使用GPU?我的笔记本电脑应该具备的最低PC要求是什么?

1 个答案:

答案 0 :(得分:4)

这完全取决于CNN的大小和深度。如果您的CNN有一个卷积层和一个完全连接的图层,并且输入图像是64x64,您将能够在合理的时间内在您的笔记本电脑上训练您的网络。如果你使用GoogLeNet数百层,并在整个ImageNet集上训练,那么即使使用视频卡也需要一周时间,所以在CPU上永远不会完成训练。

然而,对于大多数实际应用,期望具有用于训练卷积网络的GPU。请注意,在AWS上,您可以以相当合理的价格获得支持GPU的实例,特别是如果您获得现场实例,那么您不一定需要在本地拥有GPU。

最后注意:大多数框架(theano,torch,caffe,mxnet,tensorflow)允许您在CPU和GPU上执行相同的模型,只需对代码进行少量修改或不修改,因此您可以在CPU上进行本地原型设计使用一小组图像,然后当您的模型工作时,在GPU实例上在AWS上训练它。