def main()

in linux/install_gpu_driver.py [0:0]


def main():
    """
    Main function of the installation script.
    """
    args = parse_args()

    if args.action == "verify":
        if not check_driver_installed():
            print("The driver is not installed.")
        else:
            print(
                "The driver seems to be installed. Run `nvidia-smi` to check details."
            )
        return
    elif args.action == "install":
        install(args)