numpy
torch
gym # For gym examples
mpi4py # For parallelization
To install mpi4py
you need MPI
installed in the system.
Check: https://mpi4py.readthedocs.io/en/stable/install.html
A Dockerfile is provided for convenience.
pip install nes-torch --user
https://nestorch.readthedocs.io/
See https://github.com/goktug97/nes-torch/blob/master/examples
Check documentation or https://github.com/goktug97/nes-torch/blob/master/nes/config.py for parameters.
You can run the example with
PYTHONPATH="$(pwd):$PYTHONPATH" python examples/example.py
or in parallel for faster training.
PYTHONPATH="$(pwd):$PYTHONPATH" mpirun --np 2 python examples/example.py