Installing Stable Baselines

First, let's install the dependencies:

sudo apt-get update && sudo apt-get install cmake libopenmpi-dev zlib1g-dev

Several deep RL algorithms require MPI to run, so, let's install MPI:

sudo pip install mpi4py

Now, we can install Stable Baselines through pip:

pip install stable-baselines[mpi]

Note that currently, Stable Baselines works only with TensorFlow version 1.x. So, make sure you are running the Stable Baselines experiment with TensorFlow 1.x.

Now that we have installed Stable Baselines, let's see how to create our first agent using it.