- Train models on retro games
- Pit two models against each other on PvP retro games such as NHL94, Mortal Kombat or WWF Wrestlemania: The Arcade Game
- Play against an improved AI opponent
- Emulator Frontend library using Pytorch C++ to able to play against stable-retro models in apps like RetroArch
| NHL94 (1 on 1) | Wrestlemania: The Arcade game | Virtua Fighter 1 |
|---|---|---|
 |
 |
 |
Tested on Ubuntu 22.04 and Windows 11 WSL2 (Ubuntu 22.04 VM)
Requires:
- Python 3.7 and up
- stable-baselines3 with gymnasium support
- stable-retro with gymnasium support (fork of gym-retro)
sudo apt update
sudo apt-get install python3 python3-pip git zlib1g-dev libopenmpi-dev ffmpeg cmake
sudo pip3 install -U virtualenv
virtualenv --system-site-packages -p python3 ~/vretro
source ~/vretro/bin/activate
git clone https://github.com/Farama-Foundation/stable-retro.git
cd stable-retro
pip3 install -e .
pip3 install "stable_baselines3[extra]" pygame torchsummary
Windows WSL2 + Ubuntu 22.04 setup guide: https://www.youtube.com/watch?v=vPnJiUR21Og
You need to provide your own roms
In your rom directory exec this command, it will import the roms into stable-retro
python3 -m retro.import .
Currently there is two NHL 94 env in stable-retro: The original game and the '1 on 1' rom hack which as the name implies is for 1 on 1 matches instead of full teams
Note: Airstriker is a public domain rom and is already included in stable-retro
python3 model_trainer.py --env=Airstriker-Genesis --num_env=8 --num_timesteps=100_000_000 --play
