ray.runtime_context.RuntimeContext.get_accelerator_ids

RuntimeContext.get_accelerator_ids() → Dict[str, List[str]]

获取当前工作者的可见加速器ID。

返回:

一个由加速器资源名称作为键的字典。值是一个id列表 {'GPU': ['0', '1'], 'neuron_cores': ['0', '1'], 'TPU': ['0', '1']}。