gpu_app_config.yaml 517 B

12345678910111213141516
  1. base_image: {{ env["RAY_IMAGE_ML_NIGHTLY_GPU"] }}
  2. env_vars: {}
  3. debian_packages:
  4. - curl
  5. - unzip
  6. python:
  7. pip_packages:
  8. - "validators"
  9. conda_packages: []
  10. post_build_cmds:
  11. - 'rm -r wrk || true && git clone https://github.com/wg/wrk.git /tmp/wrk && cd /tmp/wrk && make -j && sudo cp wrk /usr/local/bin'
  12. - pip3 uninstall -y ray || true && pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
  13. - {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}
  14. - pip install pytest