Function print_gpu_info

python/Utilities/cuda_samples_utils.py:133–144

Print GPU device information. Parameters ---------- device : cuda.core.Device CUDA device object

(device)

def print_gpu_info(device) -> None:
    """
    Print GPU device information.

    Parameters
    ----------
    device : cuda.core.Device
        CUDA device object
    """
    print(f"Device: {device.name}")
    cc = device.compute_capability
    print(f"Compute Capability: {cc.major}.{cc.minor}")