Language

unilab.training.cuda_mps_cli.doctor¶

unilab.training.cuda_mps_cli.doctor(selector, *, root=None, cuda_visible_devices=None, run_command=<function run>, torch_module=None, uid=None, host=None)[source]¶

Validate topology and daemon state without creating or changing anything.

Parameters:

selector (str | None)
root (Path | None)
cuda_visible_devices (str | None)
run_command (Callable[..., CompletedProcess[str]])
torch_module (Any)
uid (int | None)
host (str | None)

Return type:

dict[str, Any]