ray/release/ml_user_tests/train/app_config.yaml

16 lines
402 B
YAML

base_image: "anyscale/ray-ml:nightly-py37-gpu"
env_vars:
TRAIN_PLACEMENT_GROUP_TIMEOUT_S: "2000"
debian_packages:
- curl
python:
pip_packages:
- tblib
conda_packages: [ ]
post_build_cmds:
- pip3 uninstall -y ray || true
- pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
- {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}