ray/release/xgboost_tests/cluster_gpu_small.yaml

46 lines
906 B
YAML

cluster_name: ray-xgboost-release-gpu-small
max_workers: 5
upscaling_speed: 32
idle_timeout_minutes: 15
docker:
image: anyscale/ray:latest-gpu
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: 0
max_workers: 0
gpu_1_spot:
node_config:
InstanceType: p2.xlarge
InstanceMarketOptions:
MarketType: spot
resources: {"CPU": 4, "GPU": 1}
min_workers: 4
max_workers: 4
auth:
ssh_user: ubuntu
head_node_type: cpu_4_ondemand
worker_default_node_type: gpu_1_spot
file_mounts: {
"~/xgboost_tests": "."
}
file_mounts_sync_continuously: false