Skip to content

Utils

geneva.utils.ray

get_ray_image 

get_ray_image(
    version: str,
    python_version: str,
    *,
    gpu: bool = False,
    arm: bool = False,
) -> str

Return the Ray Docker image name for the given version and options.

Parameters:

  • version (str) –

    Ray version (e.g. "2.9.0").

  • python_version (str) –

    Python version (e.g. "3.12"); dots are stripped (e.g. "3.12" → "312").

  • gpu (bool, default: False ) –

    If True, use the GPU variant of the image (adds -gpu). Default is False.

  • arm (bool, default: False ) –

    If True, use the ARM/aarch64 variant (adds -aarch64). Default is False.

Returns:

  • str –

    Image name, e.g. "rayproject/ray:2.9.0-py312" or "rayproject/ray:2.9.0-py312-gpu-aarch64".