nlsq.device.get_gpu_info

nlsq.device.get_gpu_info()[source]

Detect GPU name and SM version.

Returns:

Tuple of (gpu_name, sm_version) or (None, None) if not found. Example: (“NVIDIA GeForce RTX 4090”, 8.9)

Return type:

tuple[str | None, float | None]