gpu 利用率不高
Closed this issue · 3 comments
unlikerain commented
你好,我在训练数据时发现GPU的利用率不超过40%,请问能在哪里设置让他的可利用率高一点,以加快训练吗? 十分感谢
kuhung commented
@unlikerain 这行代码对 GPU 的显存起限定作用,但他只出现在 SSD_crop.py 中。如果代码没改动的话,不知道你的 GPU 是不是多人在共用。管理员可能对每个人的使用进行了分配限定。
config.gpu_options.per_process_gpu_memory_fraction = 0.8
kuhung commented
如果是要加快训练,可增大 batch size
unlikerain commented
好的,非常感谢