-
Notifications
You must be signed in to change notification settings - Fork 5.5k
/
agent_stress_app_config.yaml
39 lines (37 loc) · 1.54 KB
/
agent_stress_app_config.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
base_image: {{ env["RAY_IMAGE_ML_NIGHTLY_GPU"] }}
debian_packages: []
env_vars: {"RAY_INTERNAL_MEM_PROFILE_COMPONENTS": "dashboard_agent"}
debian_packages:
- htop
- curl
python:
pip_packages:
- anyscale>=0.5.47
- gcsfs==2022.5.0
- gym==0.20.0
- openskill
- protobuf>=3.15.3,<4.0.0
- pyarrow==6.0.1
- semidbm==0.5.1
- trueskill
- wandb
- memray
- typer
conda_packages: []
post_build_cmds:
- pip uninstall ale-py -y
- pip install ale-py==0.7
- pip uninstall importlib-metadata -y
- pip install importlib-metadata==4.13.0
# AutoROM downloads ROMs via torrent when they are built. The torrent is unreliable,
# so we built it for py3 and use that instead. This wheel was tested for python 3.7, 3.8,
# and 3.9.
- pip install gym[atari] https://ray-ci-deps-wheels.s3.us-west-2.amazonaws.com/AutoROM.accept_rom_license-0.5.4-py3-none-any.whl
- pip3 uninstall -y ray && pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
- pip3 install -U ray[default]
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
- sudo apt-get update
- echo "deb [signed-by=/usr/share/keyrings/cloud.google.gpg] https://packages.cloud.google.com/apt cloud-sdk main" | sudo tee -a /etc/apt/sources.list.d/google-cloud-sdk.list
- sudo apt-get install -y apt-transport-https ca-certificates gnupg
- curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | sudo apt-key --keyring /usr/share/keyrings/cloud.google.gpg add -
- sudo apt-get update && sudo apt-get install -y google-cloud-sdk