search

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.
https://rotationdetection.readthedocs.io/
Apache License 2.0
1.07k stars 180 forks source link

关于多卡训练GPU利用效率问题 #110

Closed yanglcs closed 1 year ago

yanglcs commented 2 years ago

大神你好！请教几个问题： 我在使用两块GPU训练时，一块正常，另一块GPU的利用率极低接近0，请问这是正常的么？ 是否有什么提高利用率的方法，或者增大batch_size? 如果使用8卡训练，GPU利用率是什么样的呢，是否也有这种情况呢？

yangxue0827 commented 2 years ago

这个repo的大部分方法不支持多bs，多块gpu有利用率是0我倒是没遇到过，但这个repo的利用率确实也不是很高。