有没有办法在多台笔记本电脑上训练 ML 模型？

Question

我有两台笔记本电脑，想将它们都用于 DL 模型训练。我在分布式系统方面没有任何经验，想知道是否可以同时使用两台笔记本电脑的处理能力来训练单个模型。 tf.distribute.experimental.ParameterServerStrategy 呢？有用吗？

Answer 1

是的，您可以使用多个设备来训练您的模型，并且您需要在两个设备上完成集群和工作器配置，如下所示。

tf_config = {
    'cluster': {
        'worker': ['localhost:12345', 'localhost:23456']
    },
    'task': {'type': 'worker', 'index': 0}
}

来自 Tensorflow 的 Tutorial 关于使用 Keras Multi-worker 训练 将向您展示有关配置和训练模型的所有详细信息。

希望这能回答您的问题。

is there a way to train a ML model on multiple laptops?