mirror of
https://github.com/wassname/ray.git
synced 2026-08-12 12:20:11 +08:00
[docker] Support multiple node types (#10504)
This commit is contained in:
@@ -114,3 +114,15 @@ The ``worker_setup_commands`` field (and also the ``initialization_commands`` fi
|
||||
|
||||
worker_setup_commands:
|
||||
- pip install tensorflow-gpu # Example command.
|
||||
|
||||
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.
|
||||
|
||||
.. code::
|
||||
|
||||
pull_before_run: True
|
||||
worker_image:
|
||||
- rayproject/ray-ml:latest-gpu
|
||||
worker_run_options:
|
||||
- --runtime=nvidia
|
||||
|
||||
Reference in New Issue
Block a user