如何估计不同硬件上的机器学习训练速度

数据挖掘 机器学习 硬件
2022-02-24 19:54:41

我需要提出一个购买更好的机器学习计算机的建议。有什么好的方法可以估计计算机硬件的一般训练速度吗?基本上,我想说,如果我们购买某台计算机,它在火车上的速度比我们目前的计算机快 x 倍。目前我有一个 i5 63000U 2.4ghz 2 核 cpu,没有 gpu。我知道切换到带有 gpu 的计算机会大大加快速度。我只需要一种方法来量化改进以证明投资的合理性。任何想法将不胜感激。提前致谢。

1个回答

有什么好的方法可以估计计算机硬件的一般训练速度吗?

可能不是一般的,因为“训练速度”意味着如果你训练一个 kNN 模型(几乎没有训练)或一个复杂的 DL 模型,什么样的数据和什么样的任务完全不同。

然而,对于固定设置存在估计,例如我发现了这个 MLPerf Benhchmark这篇论文

顺便说一句,台式计算机并不适合计算机密集型机器学习,集群或云计算提供了更好的选择。