wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 03:34:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add accelerator-type to multi node type example YAML (#10871)

Browse Source
This commit is contained in:
Eric Liang
2020-09-17 17:09:35 -07:00
committed by Barak Michener
parent 2fb29eb680
commit 1d520bf796
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/autoscaler/aws/example-multi-node-type.yaml
+3 -3
View File
@@ -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.