Source code for track.utils.system



[docs]def get_gpu_name():
    try:
        import torch
        current_device = torch.cuda.current_device()
        return torch.cuda.get_device_name(current_device)
    except ImportError:
        return None