PJLab-ADG/DriveArena

Hardware configuration

Closed this issue · 5 comments

What if I just want to reproduce the inference process of your model and not train a GPU that requires much memory?

For inference only, a 4090-level GPU should be sufficient.

Can a 4090 class GPU run your World Dreamer, UniAD and Traffic Manager in its entirety?

Yes, a RTX 4090 GPU would be efficient for running inference on both World Dreamer and UniAD.
Traffic Manager, doesn't require GPU resources and should be run on your local machine with a display.

Okay, thank you very much for your reply.

Yes, a RTX 4090 GPU would be efficient for running inference on both World Dreamer and UniAD.
Traffic Manager, doesn't require GPU resources and should be run on your local machine with a display.

hello i wonder how much gpu memory need for training dreamer?