用gpu服务器训练模型时,发现显存满了,但gpu利用率很低,且周期性变化,比如0-25-45,都不高,同样的代码在另一台服务器上,利用率是90多,且模型训练速度也很低,迭代起来一卡一卡的。
在dataloader中增加num_workers什么的,都不管用,最后发现,我加载的数据放到home(sda)下了,正确应该放到另一个移动硬盘中(sdb)
如果你和我的情况一致,建议你复制数据到sdb中,同时修改dataloader中的数据目录,修改后,我的服务器gpu利用率直接到90多,且速度很快。
用gpu服务器训练模型时,发现显存满了,但gpu利用率很低,且周期性变化,比如0-25-45,都不高,同样的代码在另一台服务器上,利用率是90多,且模型训练速度也很低,迭代起来一卡一卡的。
在dataloader中增加num_workers什么的,都不管用,最后发现,我加载的数据放到home(sda)下了,正确应该放到另一个移动硬盘中(sdb)
如果你和我的情况一致,建议你复制数据到sdb中,同时修改dataloader中的数据目录,修改后,我的服务器gpu利用率直接到90多,且速度很快。