gpu_select

class GPUSelect

Bases: object

canary_gpu_backend_detect(config: Config) → str | None

canary_gpu_list_gpus(config: Config) → list[dict] | None

canary_addoption(parser: Parser) → None

canary_addhooks(pluginmanager: CanaryPluginManager) → None

detect_gpu_backend(config: Config) → None

canary_fill_gpu(config: Config, pool: dict[str, dict[str, Any]]) → None