我想使用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
. . .
学习代码像这样简短地开始。
如何同时使用0和1 GPU?
我想将GPU使用率从35%提高到99%。我该怎么办?
我搜索了很多次,以至于找不到方法。对不起。 感谢您的帮助。