wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-12 12:51:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Docker] Automagically add "runtime=nvidia" (#11125)

Browse Source
This commit is contained in:
Ian Rodney
2020-10-01 17:04:19 -07:00
committed by GitHub
parent 681c24754a
commit 0d5b09f426
10 changed files with 21 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/autoscaler/aws/example-full.yaml
+1 -2
View File
@@ -32,8 +32,7 @@ docker:
# Example of running a GPU head with CPU workers
# head_image: "rayproject/ray:0.8.7-gpu"
# head_run_options:
# - --runtime=nvidia
# Allow Ray to automatically detect GPUs
# worker_image: "rayproject/ray:0.8.7"
# worker_run_options: []
python/ray/autoscaler/aws/example-gpu-docker.yaml
-5
View File
@@ -25,16 +25,11 @@ autoscaling_mode: default
docker:
image: "rayproject/ray:0.8.7-gpu"
container_name: "ray-nvidia-docker-test" # e.g. ray_docker
run_options:
- --runtime=nvidia
# # Example of running a GPU head with CPU workers
# head_image: "rayproject/ray:0.8.7-gpu"
# head_run_options:
# - --runtime=nvidia
# worker_image: "rayproject/ray:0.8.7"
# worker_run_options: []
# The autoscaler will scale up the cluster to this target fraction of resource
# usage. For example, if a cluster of 10 nodes is 100% busy and
python/ray/autoscaler/aws/example-ml.yaml
-3
View File
@@ -37,11 +37,8 @@ docker:
# Example of running a GPU head with CPU workers
# head_image: "rayproject/ray:0.8.7-gpu"
# head_run_options:
# - --runtime=nvidia
# worker_image: "rayproject/ray:0.8.7"
# worker_run_options: []
# The autoscaler will scale up the cluster to this target fraction of resource
# usage. For example, if a cluster of 10 nodes is 100% busy and