wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-09 11:32:43 +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
doc/source/cluster/autoscaling.rst
+1 -3
View File
@@ -134,12 +134,10 @@ The ``worker_setup_commands`` field (and also the ``initialization_commands`` fi
Docker Support
~~~~~~~~~~~~~~
The ``worker_image`` and ``pull_before_run`` fields override the correpsonding field in the top level ``docker`` section for the node type. The ``worker_run_options`` field is combined with top level ``docker: run_options`` field to produce the docker run command for the given node_type. The following configuration is for a GPU enabled node type.
The ``worker_image`` and ``pull_before_run`` fields override the correpsonding field in the top level ``docker`` section for the node type. The ``worker_run_options`` field is combined with top level ``docker: run_options`` field to produce the docker run command for the given node_type. The following configuration is for a GPU enabled node type. Ray will automatically select the Nvidia docker runtime if it is available.
.. code::
pull_before_run: True
worker_image:
- rayproject/ray-ml:latest-gpu
worker_run_options:
- --runtime=nvidia