Skip to main content
 首页 » 编程设计

python之同时使用 2 个 GPU 进行不同的 TensorFlow\Python

2025年02月15日25unruledboy

  1. Python 2.7
  2. tensorflow
  3. Ubuntu

你好我亲爱的 friend 们,有问题。是否可以使用多 GPU 同时训练不同的模型?通过简单的代码操作将它们用于一次训练不是问题,但是如果我想尝试同时使用不同的参数同时进行其中的 2 次训练怎么办? 对不起,如果这个问题很简单,是编程和 TensorFlow 的新问题。在指南 Using GPU's我试图找到一个关于它的问题,但据我所知,这里仅针对一个过程。

请您参考如下方法:

看起来有可能。对于您的模型,您使用“with tf.device..”并为不同的模型指定不同的 gpus。然后运行这两个程序,看看会发生什么。我在不同的 gps 上尝试了一个简单的程序,它运行了。让我们知道您找到的输出。