我尝试在Labtop中使用4 GPU。所以我使用命令
new_model = multi_gpu_model(model,gpus = 4)
效果很好,但是有一个问题。
只有第一个GPU具有更多的GPU内存。
以下是我的代码和watch nvidia-smi
检查的我的GPU内存的状态
仅当在命令new_model = multi_gpu_model(model, gpus=4)
之前组成模型时,第一个gpu0才增加
那时,第一个gpu0中出现4000MiB内存。
为什么会出现这种不公平的GPU使用情况? 而我该如何平均使用4个GPU? 请给我提示。
非常感谢。