import torch | |
device = torch.accelerator.current_accelerator() or torch.device('cpu') | |
print(f"Device set to {torch.Tensor().to(device).device}") | |
__all__ = ['device'] | |
import torch | |
device = torch.accelerator.current_accelerator() or torch.device('cpu') | |
print(f"Device set to {torch.Tensor().to(device).device}") | |
__all__ = ['device'] | |