import torch

device = torch.accelerator.current_accelerator() or torch.device('cpu')

print(f"Device set to {torch.Tensor().to(device).device}")

__all__ = ['device']