PlaNet: A Deep Planning Network for Reinforcement Learning [1]. Supports symbolic/visual observation spaces. Supports some Gym environments (including classic control/non-MuJoCo environments, so DeepMind Control Suite/MuJoCo are optional dependencies).
Run with python.main.py
. For best performance with DeepMind Control Suite, try setting environment variable MUJOCO_GL=egl
(see instructions and details here).
- Python 3
- DeepMind Control Suite (optional)
- Gym
- OpenCV Python
- Plotly
- PyTorch
To install all dependencies with Anaconda run conda env create -f environment.yml
and use source activate rainbow
to activate the environment.