waymo-research/waymax

Minimum system requirement

mshah72 opened this issue · 1 comments

mshah72 commented

What is the minimum system requirement for this simulator?

Is it possible to try this using Google Colab?

justinjfu commented

Hi,

We haven't done extensive testing on different machines for minimum requirements. A modern GPU or TPU will speed up simulation significantly over any CPU (e.g. NVidia 3080+), and more GPU/TPU memory is helpful for larger batch sizes.

Google colab should work, just run !pip install git+https://github.com/waymo-research/waymax.git@main#egg=waymo-waymax at the top of the colab. If you are a researcher this is excellent for being able to take advantage of the TPUs/GPUs that come with colab.