DDP: out of CUDA memory

XinzeLee / RotateObjectDetection

This repository is based on Ultralytics/yolov5, with adjustments to enable rotate prediction boxes.

113 stars 21 forks source link

Open Iipython opened 2 years ago

Iipython commented 2 years ago

我是用了5,6两张卡，发现多次报错 out of CUDA memory. 后来我发现两张卡上现存分布是不均衡的，导致一张卡很容易爆炸 KP2GB%J$RME))TT6W0{UN7Y 。请问原因可能是什么？