From 1d520bf7967e8b4ee38b969da9cf1a06be960c8b Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Thu, 17 Sep 2020 17:09:35 -0700 Subject: [PATCH] Add accelerator-type to multi node type example YAML (#10871) --- python/ray/autoscaler/aws/example-multi-node-type.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/ray/autoscaler/aws/example-multi-node-type.yaml b/python/ray/autoscaler/aws/example-multi-node-type.yaml index ec3d5c37b..008337de2 100644 --- a/python/ray/autoscaler/aws/example-multi-node-type.yaml +++ b/python/ray/autoscaler/aws/example-multi-node-type.yaml @@ -28,14 +28,14 @@ available_node_types: gpu_1_ondemand: node_config: InstanceType: p2.xlarge - resources: {"CPU": 4, "GPU": 1, "Custom2": 2} + resources: {"CPU": 4, "GPU": 1, "Custom2": 2, "accelerator_type:K80": 1} max_workers: 4 worker_setup_commands: - pip install tensorflow-gpu # Example command. gpu_8_ondemand: node_config: - InstanceType: p2.8xlarge - resources: {"CPU": 32, "GPU": 8} + InstanceType: p3.8xlarge + resources: {"CPU": 32, "GPU": 4, "accelerator_type:V100": 1} max_workers: 2 worker_setup_commands: - pip install tensorflow-gpu # Example command.