ray/release/rllib_tests/app_config.yaml

22 lines
548 B
YAML
Executable file

base_image: "anyscale/ray-ml:1.3.0-gpu"
env_vars: {}
debian_packages:
- libglib2.0-0
- unrar
python:
pip_packages:
- gym[atari]
- atari_py
- pybullet
conda_packages: []
post_build_cmds:
- pip uninstall -y numpy ray || true
- sudo rm -rf /home/ray/anaconda3/lib/python3.7/site-packages/numpy
- pip3 install numpy==1.19.5 || true
- pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
# Install Atari ROMs.
- wget http://www.atarimania.com/roms/Roms.rar
- unrar x Roms.rar
- python -m atari_py.import_roms .