Installing Stable Baselines
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 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.