如何使用多GPU?

时间:2019-05-09 01:44:34

标签: tensorflow gpu multi-gpu

我想使用tensorflow multi gpu。当前,有两个GPU #0#1,如果您运行nvidia-smi并看到GPU使用率,则只有#0 GPU被使用约35%。我想知道如何同时使用99%的0#1#

此代码在学习代码之前添加并执行。我不知道如何同时使用0和1,所以我只使用0,但只有35%。

tf.device("/device:GPU:0")

    config=tf.ConfigProto()
    config.gpu_options.allow_growth = True
    config.gpu_options.per_process_gpu_memory_fraction = 0.9       

    with tf.Session(config=config) as sess :
        while(True) :
            prev_cost = 9999999999
            break_condition_count = 0
. . .

学习代码像这样简短地开始。

  1. 如何同时使用0和1 GPU?

  2. 我想将GPU使用率从35%提高到99%。我该怎么办?

我搜索了很多次,以至于找不到方法。对不起。 感谢您的帮助。

0 个答案:

没有答案