Installation

Install from PyPi

You can install VMAS from PyPi.

pip install vmas

Install from source

If you want to install the current main version (more up to date than latest release), you can do:

git clone https://github.com/proroklab/VectorizedMultiAgentSimulator.git
cd VectorizedMultiAgentSimulator
pip install -e .

Install optional requirements

By default, vmas has only the core requirements. Here are some optional packages you may want to install.

Wrappers

If you want to use VMAS environment wrappers, you may want to install VMAS with the following options:

# install gymnasium for gymnasium wrapper
pip install vmas[gymnasium]

# install rllib for rllib wrapper
pip install vmas[rllib]

Training

You may want to install one of the following training libraries

pip install benchmarl
pip install torchrl
pip install "ray[rllib]"==2.1.0 # We support versions "ray[rllib]<=2.2,>=1.13"

Utils

You may want to install the following additional tools

# install rendering dependencies
pip install vmas[render]
# install testing dependencies
pip install vmas[test]