uqlm.utils.device.get_best_device#
- uqlm.utils.device.get_best_device()#
Detects and returns the best available PyTorch device. Prioritizes CUDA (NVIDIA GPU), then MPS (macOS), then CPU.
- Return type:
device- Returns:
The best available device.
- Return type:
torch.device
Examples
>>> device = get_best_device() >>> print(f"Using {device.type} device")