Fatal编程技术网
  • machine-learning/
  • Machine learning 我可以同时在2个不同的GPU上训练2个不同的NN吗？

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

machine-learning tensorflow

Machine learning 我可以同时在2个不同的GPU上训练2个不同的NN吗？,machine-learning,tensorflow,gpu,tensorflow-gpu,Machine Learning,Tensorflow,Gpu,Tensorflow Gpu,我从未有过2个GPU，但我很快就会在一台PC上有2个GPU 有时我想用这两张卡训练一个模型，以便尽可能快地训练。然而，我想象一种情况，我想同时训练两个不同的模型 是否可以使用Tensorflow或PyTorch这样的框架？您可以使用非常简单的方法。启动两个不同的python脚本，一个使用第一个模型，另一个使用第二个模型。在TensorFlow中，您可以使用tf.device“/gpu:0”和tf.device“/gpu:1”等显式指定设备。有关详细信息，请参阅文档

我从未有过2个GPU，但我很快就会在一台PC上有2个GPU

有时我想用这两张卡训练一个模型，以便尽可能快地训练。然而，我想象一种情况，我想同时训练两个不同的模型

是否可以使用Tensorflow或PyTorch这样的框架？

您可以使用非常简单的方法。启动两个不同的python脚本，一个使用第一个模型，另一个使用第二个模型。在TensorFlow中，您可以使用tf.device“/gpu:0”和tf.device“/gpu:1”等显式指定设备。有关详细信息，请参阅文档