- Python 3.8+
- Specific versions of pip and setuptools due to a bug with gym:
pip install -U setuptools==65.5.0 pip==21
- (on M1 Macs) you need to set environment variables due to a bug in grpcio:
export GRPC_PYTHON_BUILD_SYSTEM_OPENSSL=1
export GRPC_PYTHON_BUILD_SYSTEM_ZLIB=1
- (Optional) OpenGL (to render gym environments)
- (Optional) FFmpeg (to encode videos of renders)
- (Optional) MuJoCo (follow instructions to install mujoco_py v1.5 here)
To install the latest PyPI release, simply run:
pip install imitation
Installation from source is useful if you wish to contribute to the development of imitation
, or if you need features that have not yet been made available in a stable release:
git clone http://github.com/HumanCompatibleAI/imitation
cd imitation
pip install -e .
There are also a number of dependencies used for running tests and building the documentation, which can be installed with:
pip install -e ".[dev]"