Function main

python/1_GettingStarted/systemInfo/systemInfo.py:171–205

()

def main() -> int:
    import argparse

    parser = argparse.ArgumentParser(
        description="Print CUDA system / NVML information via cuda.core.system"
    )
    parser.add_argument(
        "--no-topology",
        action="store_true",
        help="Skip cross-device topology/P2P queries",
    )
    args = parser.parse_args()

    print_driver_info()

    num_devices = system.get_num_devices()
    print_header(f"Devices detected: {num_devices}")
    if num_devices == 0:
        print("No CUDA-capable devices found.")
        return 0
    if not CUDA_BINDINGS_NVML_IS_COMPATIBLE:
        print(
            "NVML is not compatible with the installed driver; skipping device detail."
        )
        return 0

    devices = [system.Device(index=i) for i in range(num_devices)]
    for device in devices:
        print_device_info(device)

    if not args.no_topology:
        print_topology(devices)

    print("\nDone")
    return 0


if __name__ == "__main__":