如何获取CuPy的所有可用设备？

Question

我怎么才能获取CuPy支持的所有设备呢？我想为CuPy写一个类似的版本：

if is_torch(xp):
    devices = ['cpu']
    import torch  # type: ignore[import]
    num_cuda = torch.cuda.device_count()
    for i in range(0, num_cuda):
        devices += [f'cuda:{i}']
    if torch.backends.mps.is_available():
        devices += ['mps']
    return devices