ModuleNotFoundError: No module named 'baselines.common'

Question

I ran the following code in Google Colab:

    from baselines.common.atari_wrappers import make_atari, wrap_deepmind

And got the following error:

    ModuleNotFoundError: No module named 'baselines.common'

I tried !pip install baselines, but it didn't fixed the issue.....

Answer 1

try following instruction:

!apt install swig cmake libopenmpi-dev zlib1g-dev
!pip install stable-baselines[mpi]==2.8.0 box2d box2d-kengz

from