Keras Multi GPU内存使用情况不同

时间:2019-10-16 12:17:49

标签: keras gpu equals

我尝试在Labtop中使用4 GPU。所以我使用命令

  

new_model = multi_gpu_model(model,gpus = 4)

效果很好,但是有一个问题。 只有第一个GPU具有更多的GPU内存。 以下是我的代码和watch nvidia-smi检查的我的GPU内存的状态 enter image description here

enter image description here

仅当在命令new_model = multi_gpu_model(model, gpus=4)之前组成模型时,第一个gpu0才增加

那时,第一个gpu0中出现4000MiB内存。

为什么会出现这种不公平的GPU使用情况? 而我该如何平均使用4个GPU? 请给我提示。

非常感谢。

0 个答案:

没有答案