agent_stress_app_config.yaml 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. base_image: {{ env["RAY_IMAGE_NIGHTLY_GPU"] | default("anyscale/ray-ml:2.0.0-py37-gpu") }}
  2. debian_packages: []
  3. env_vars: {"RAY_INTERNAL_MEM_PROFILE_COMPONENTS": "dashboard_agent"}
  4. debian_packages:
  5. - htop
  6. - curl
  7. python:
  8. pip_packages:
  9. - anyscale>=0.5.47
  10. - gcsfs==2022.5.0
  11. - gym==0.20.0
  12. - openskill
  13. - protobuf>=3.15.3,<4.0.0
  14. - pyarrow==6.0.1
  15. - semidbm==0.5.1
  16. - trueskill
  17. - wandb
  18. - memray
  19. - typer
  20. conda_packages: []
  21. post_build_cmds:
  22. - pip uninstall ale-py -y
  23. - pip install ale-py==0.7
  24. - pip uninstall importlib-metadata -y
  25. - pip install importlib-metadata==4.13.0
  26. # AutoROM downloads ROMs via torrent when they are built. The torrent is unreliable,
  27. # so we built it for py3 and use that instead. This wheel was tested for python 3.7, 3.8,
  28. # and 3.9.
  29. - 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
  30. - pip3 uninstall -y ray && pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
  31. - pip3 install -U ray[default]
  32. - {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
  33. - sudo apt-get update
  34. - 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
  35. - sudo apt-get install -y apt-transport-https ca-certificates gnupg
  36. - curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | sudo apt-key --keyring /usr/share/keyrings/cloud.google.gpg add -
  37. - sudo apt-get update && sudo apt-get install -y google-cloud-sdk