在普通个人计算机上训练机器学习模型(带有图像输入)是否可行?

人工智能 神经网络 机器学习 图像识别 训练 计算机视觉
2021-11-06 07:13:11

有很多机器学习系统的例子可以识别物体并以非常高的精度从图像中提取其他信息。训练这种系统的模型是必要的(我猜)一台具有强大计算能力的计算机。

我的问题是:对于以图像作为输入的系统,根据问题的复杂性,在普通笔记本电脑上训练模型是否可行?会花很多时间吗?

我知道训练任何机器学习模型所花费的时间将取决于许多变量。我真的不期望这里有一个定量的答案,我只是想知道我是否会被迫升级我的计算机来开发和训练以图像为输入的机器学习模型。

1个回答

你可以在一台普通的笔记本电脑上玩,但训练会很慢,而且你会受到模型大小的限制。
一旦您尝试构建更严重的东西,您将很快耗尽内存。如果你真的想做图像识别之类的事情,建议使用带有 GPU 的系统。如果您购买任何东西,我不会选择任何内存小于 8 GB 的 GPU,也不会选择 X99 主板,以保持稍后在需要时以全速添加第二个 GPU 的灵活性。
不要混淆在手机等上运行的图像识别。这些是经过训练的模型,实际上只是被应用,这是一个更容易的过程。培训要贵得多。