【问题标题】:Using 2 GPU at same time for different traning's TensorFlow\Python同时使用 2 个 GPU 进行不同训练的 TensorFlow\Python
【发布时间】:2017-02-03 02:08:38
【问题描述】:
  1. Python 2.7
  2. TensorFlow
  3. Ubuntu

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

【问题讨论】:

    标签: python tensorflow gpu


    【解决方案1】:

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

    【讨论】:

    • 是的,这是正确的。 Tensorflow 在同一进程中支持多个 GPU。您可以使用with tf.device("GPU:1") 将算子分配给 GPU 1。
    猜你喜欢
    • 1970-01-01
    • 2018-12-02
    • 1970-01-01
    • 2018-03-03
    • 2021-04-24
    • 2021-10-03
    • 2018-04-06
    • 1970-01-01
    • 2019-05-01
    相关资源
    最近更新 更多