Function get_current_device_name

lightllm/utils/device_utils.py:80–88

()

@lru_cache(maxsize=None)
def get_current_device_name():
    import torch

    if torch.cuda.is_available():
        device = torch.cuda.current_device()
        gpu_name = torch.cuda.get_device_name(device).replace(" ", "_")
        return gpu_name
    else:
        return None


@lru_cache(maxsize=None)