Skip to content

Allow allocation of multiple GPUs. #1

@rshin

Description

@rshin

Use https://docs.nvidia.com/deploy/nvml-api/group__nvmlDeviceStructs.html#group__nvmlDeviceStructs_1g5fd9890987e4b5768817dd64438e24f4 to preferentially choose GPUs with good communication between them.

When allocating single GPUs, try to leave room for multi-GPU allocation.

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions