ray/release/xgboost_tests/cluster_cpu_moderate.yaml
Kai Fricke a0f73cf3f7
[xgboost] Update XGBoost release test configs (#13941)
* Update XGBoost release test configs

* Use GPU containers

* Fix elastic check

* Use spot instances for GPU

* Add debugging output

* Fix success check, failure checking, outputs, sync behavior

* Update release checklist, rename mounts
2021-02-17 23:00:49 +01:00

41 lines
751 B
YAML

cluster_name: ray-xgboost-release-cpu-moderate
max_workers: 32
upscaling_speed: 32
idle_timeout_minutes: 15
docker:
image: anyscale/ray:latest
container_name: ray_container
pull_before_run: true
provider:
type: aws
region: us-west-2
availability_zone: us-west-2a
cache_stopped_nodes: false
available_node_types:
cpu_4_ondemand:
node_config:
InstanceType: m5.xlarge
resources: {"CPU": 4}
min_workers: 31
max_workers: 31
auth:
ssh_user: ubuntu
head_node_type: cpu_4_ondemand
worker_default_node_type: cpu_4_ondemand
file_mounts: {
"~/xgboost_tests": "."
}
file_mounts_sync_continuously: false
setup_commands:
- /bin/bash ~/xgboost_tests/setup_xgboost.sh