我可以同时在2个不同的GPU上训练2个不同的NN吗？

Question

我从来没有2个GPU，但我很快就会在一台PC上运行2个。

有时我想用两张牌训练模型，尽快训练。但是我想象一种情况，我想要同时训练两个不同的模型。

是否可以使用Tensorflow或PyTorch等框架？

Answer 1

您可以使用非常简单的方法。启动两个不同的python脚本，一个是第一个模型，另一个是第二个。在TensorFlow中，您可以使用with tf.device('/gpu:0'):，with tf.device('/gpu:1'):等明确指定设备。有关详细信息，请查看文档TensorFlow Using GPUs。