仅使用CPU训练神经网络

时间:2019-02-27 20:07:12

标签: neural-network deep-learning computer-vision conv-neural-network

我正在远程服务器上的虚拟机上工作,我想在其上训练神经网络,但是我没有在此VM中使用GPU。是否可以仅使用CPU在此VM上训练网络?如果是这样,那么它是否适用于大型数据集?还是会有问题?

1 个答案:

答案 0 :(得分:2)

我已经使用Tensorflow来训练深度神经网络。我已经在 GPU 仅CPU 中使用了它。我其余的回答是在 Tensorflow 的背景下进行的。

请注意,卷积神经网络通常比标准的常规前馈神经网络更耗费资源,因为CNN处理的维数要大得多。如果您不使用深层的CNN,则可以使用CPU来限制较小的数据集。

在我的情况下,最初我只使用CPU进行训练,然后由于速度提高而转入GPU模式。

速度示例

在不到15分钟的时间内使用GPU,我就能够训练整个MNIST。在CPU上进行培训的速度要慢得多,但是您仍然可以通过减少培训数据集的大小来学习。

带有GPU的Tensorflow

https://www.tensorflow.org/install/gpu 您将需要完成所有安装步骤。这不仅涉及安装Tensorflow,而且涉及CUDA库。

什么是CUDA?

CUDA是NVIDIA开发的用于使用GPU编程的规范。他们提供了与底层硬件对话的本机库。 https://docs.nvidia.com/cuda/

How to use TensorFlow GPU?