Skip to content

Commit 361fab9

Browse files
committed
ENG-26189: Replaces deprecated parameters
1 parent 0ecefdd commit 361fab9

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

modules/running-distributed-data-science-workloads-from-ds-pipelines.adoc

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -93,9 +93,10 @@ def ray_fn():
9393
namespace="my_project", <4>
9494
name="raytest",
9595
num_workers=1,
96-
head_cpus="500m",
97-
min_memory=1,
98-
max_memory=1,
96+
head_cpu_requests="500m",
97+
head_cpu_limits="500m",
98+
worker_memory_requests=1,
99+
worker_memory_limits=1,
99100
worker_extended_resource_requests={“nvidia.com/gpu”: 1}, <5>
100101
image="quay.io/modh/ray:2.35.0-py39-cu121", <6>
101102
local_queue="local_queue_name", <7>
@@ -159,8 +160,6 @@ if __name__ == '__main__': <14>
159160
<5> Optional: Specifies the requested accelerators for the Ray cluster (in this example, 1 NVIDIA GPU).
160161
If you do not use NVIDIA GPUs, replace `nvidia.com/gpu` with the correct value for your accelerator; for example, specify `amd.com/gpu` for AMD GPUs.
161162
If no accelerators are required, set the value to 0 or omit the line.
162-
Note: To specify the requested accelerators for the Ray cluster, use the `worker_extended_resource_requests` parameter instead of the deprecated `num_gpus` parameter.
163-
For more details, see the link:https://github.com/project-codeflare/codeflare-sdk/blob/v0.18.0/src/codeflare_sdk/cluster/config.py#L43-L73[CodeFlare SDK documentation].
164163
<6> Specifies the location of the Ray cluster image.
165164
The Python version in the Ray cluster image must be the same as the Python version in the workbench.
166165
If you omit this line, one of the default CUDA-compatible Ray cluster images is used, based on the Python version detected in the workbench.

0 commit comments

Comments
 (0)