environment-gpu.yml 295 Bytes
Newer Older
1
name: flatland-baseline-gpu-env
2 3 4 5 6
dependencies:
    - python=3.7
    - tensorflow-gpu=2.1.0
    - pip=20.0
    - pip:
7
        - flatland-rl==2.2.1
8 9 10 11
        - CairoSVG==2.4.2
        - pyhumps==1.3.1
        - gputil==1.4.0
        - pyhumps==1.3.1
12
        - wandb==0.9.2
MasterScrat's avatar
MasterScrat committed
13
        - ray[rllib]==0.8.5