search

OpenDriveLab / UniAD

[CVPR 2023 Best Paper Award] Planning-oriented Autonomous Driving
Apache License 2.0
3.57k stars 406 forks source link

GPU and Memory #186

Open MPCheng-ZW opened 6 months ago

MPCheng-ZW commented 6 months ago

Thanks for your excellent work.

Can I run the code on a computer equipped with a single NVIDIA RTX 4090 GPU?

Additionally, how much disk memory is required to ensure successful running of this code?

Thanks again.

ZXIAOLIANG commented 5 months ago

Hi, have you made progress on running on a single 4090 GPU? Any help or guide would be appreciated!

MPCheng-ZW commented 5 months ago

Hi, have you made progress on running on a single 4090 GPU? Any help or guide would be appreciated!

hello，我前段时间在单个3090上跑了一下，用的small版本的dataset。 Stage1: Perception training 跑的时候out of memory Stage2: End-to-end training 可以run