Installation

Install TorchRL

You can install TorchRL from PyPi.

pip install torchrl

For more details, or for installing nightly versions, see the TorchRL installation guide.

You can just install it from PyPi

pip install benchmarl

Or also clone it locally to access the configs and scripts

git clone https://github.com/facebookresearch/BenchMARL.git
pip install -e BenchMARL

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

You may want to install the following rendering and logging tools

pip install wandb moviepy torchvision av

All environment dependencies are optional in BenchMARL and can be installed separately.

pip install vmas

pip install "pettingzoo[all]"

pip install "dm-meltingpot"

Follow the instructions on the environment repository.

Here is how we install it on linux.

Some models in BenchMARL require extra dependencies that can be installed separately

GNN models require pytorch_geometric.

To install it, you can run:

pip install torch_geometric

For more information, see the installation instructions of the library.