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.