when training Transfusion_LC, CUDA out of memory. （NVIDIA 2080 8G）

Open wangyd-0312 opened this issue 3 years ago • 3 comments

hi, Mr. Bai I have trained Transfusion_L, and got the fusion_model.pth. When I want to train Transfusion_LC, I get this error "CUDA out of memory", I use NVIDIA 2080 8G, and I change sample_per_gpu=1、lr=0.0025, and the database is nuscenes mini. I want to know if 2080 8G can run the Transfusion_LC?

Looking forward your replay, thank you. 捕获1

Jun 28 '22 08:06 wangyd-0312

Hi, I am afraid 8G is not enough for training TransFusionLC, you should find GPUs with larger memory (i.e. 16G should be enough for LC).

Jun 30 '22 01:06 XuyangBai

Thank you for your replay!!

Jun 30 '22 06:06 wangyd-0312

@XuyangBai is a single GPU sufficient for training L and LC? Or is distributed training across many 16G GPUs required?

Mar 23 '23 11:03 wqueree