/traffic-intersection-rl-environment-cli

custom traffic intersection environment for reinforcement learning (without gui)

Primary LanguagePython

traffic-intersection-rl-environment-cli

Meeting Requirements

pip install -r requirements.txt

installing environment

  • cd custom_gym

  • pip install -e .

  • cd ..

Start training and see results

python train.py

View in SUMO

python traffic.py

For gpu support

You need to uninstall pytorch cpu version if installed already and install the gpu version.

Problem with requirements.txt ?

You may change the version or the name of the dependency to match your platform.