search

motional / nuplan-devkit

The devkit of the nuPlan dataset.
https://www.nuplan.org
Other
662 stars 126 forks source link

How to run simulation with GPUs? #301

Closed jchengai closed 1 year ago

jchengai commented 1 year ago

I found that the default simulation setting uses all CPU cores. How to properly set number_of_gpus_allocated_per_simulation , for example, for a single node machine with 4 GPUs?

patk-motional commented 1 year ago

Hi @jchengai,

We have not explored running with multiple GPUs on a single machine. I've mentioned in this issue that setting worker=sequential is the easiest way to get GPU access.