def METHOD_NAME(self, device: torch.device) -> None:
    """
    Raises:
        MisconfigurationException:
            If the selected device is not GPU.
    """
    if device.type != "cuda":
        raise MisconfigurationException(f"Device should be GPU, got {device} instead")
    _check_cuda_matmul_precision(device)
    torch.cuda.set_device(device)