Commit 64dd8fa0 authored by MasterScrat's avatar MasterScrat

Updated requirements.txt, added environments.yml

parent 27e8d3bc
......@@ -15,28 +15,35 @@ All files are stored in a experiment folder under `experiments/<env-name>/<exper
Setup
---
Tested with Python 3.6 and 3.7
Using conda (recommended):
```
conda create --name ray-env python=3.7 --yes
# with GPU support:
conda env create -f environment-gpu.yml
# or, without GPU support:
#conda env create -f environment-cpu.yml
conda activate flatland-env
pip install -r requirements.txt
```
You may need to install/update bazel: [Ubuntu guide](https://docs.bazel.build/versions/master/install-ubuntu.html)
Using pip:
```
pip install ray[rllib]
pip install tensorflow
# with GPU support
# conda install -c anaconda tensorflow-gpu
# no GPU support:
pip install -r requirements.txt
```
You may need to install/update bazel: [Ubuntu guide](https://docs.bazel.build/versions/master/install-ubuntu.html)
## Usage
Training example:
`python ./train.py -f experiments/flatland_random_sparse_small/global_obs_conv_net/ppo.yaml`
Test example:
Evaluation example:
`python ./rollout.py /tmp/ray/checkpoint_dir/checkpoint-0 --run PPO --no-render
--config '{"env_config": {"test": true}}' --episodes 1000 --out rollouts.pkl`
......
name: flatland-env
dependencies:
- python=3.7
- pip:
- flatland-rl==2.1.10
- CairoSVG==2.4.2
# - pycairo==1.19.1 # fails!
- pyhumps==1.3.1
- gputil==1.4.0
- pyhumps==1.3.1
- wandb==0.8
- ray[rllib]==0.8.4
- tensorflow==2.1.0
\ No newline at end of file
name: flatland-env
dependencies:
- python=3.7
- pip=20.0
- pip:
- git+https://gitlab.aicrowd.com/flatland/flatland.git
- CairoSVG==2.4.2
# - pycairo==1.19.1 # fails!
- pyhumps==1.3.1
- gputil==1.4.0
- pyhumps==1.3.1
- wandb==0.8
- ray[rllib]==0.8.4
- tensorflow==2.1.0
\ No newline at end of file
name: flatland-env
dependencies:
- python=3.7
- tensorflow-gpu=2.1.0
- pip=20.0
- pip:
- git+https://gitlab.aicrowd.com/flatland/flatland.git
- CairoSVG==2.4.2
# - pycairo==1.19.1 # fails!
- pyhumps==1.3.1
- gputil==1.4.0
- pyhumps==1.3.1
- wandb==0.8
- ray[rllib]==0.8.4
\ No newline at end of file
name: flatland-env
dependencies:
- python=3.7
- tensorflow-gpu=2.1.0
- pip=20.0
- pip:
- flatland-rl==2.1.10
- CairoSVG==2.4.2
# - pycairo==1.19.1 # fails!
- pyhumps==1.3.1
- gputil==1.4.0
- pyhumps==1.3.1
- wandb==0.8
- ray[rllib]==0.8.4
\ No newline at end of file
CairoSVG==2.4.2
flatland-rl==2.1.10
pycairo==1.19.1
CairoSVG==2.4.2
#pycairo==1.19.1 # fails!
pyhumps==1.3.1
gputil==1.4.0
pyhumps==1.3.1
wandb==0.8.34
\ No newline at end of file
wandb==0.8
ray[rllib]==0.8.4
tensorflow==2.1.0
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment