def install_driver_runfile()

in linux/install_gpu_driver.py [0:0]

• 27 lines of code
• 10 McCabe index (conditional complexity)

def install_driver_runfile(system: System, version: str):
    dkms = "--dkms"
    if system in (System.RHEL, System.Rocky) and version.startswith("8"):
        # There is a problem with DKMS installation on Rocky and RHEL 8
        # DKMS is disabled for those systems, you'll need to reinstall the drivers
        # with every kernel update.
        dkms = ""

    if detect_gpu_device() == TESLA_K80_DEVICE_CODE:
        run(f"curl -fSsl -O {K80_DRIVER_URL}")
        binary = f"NVIDIA-Linux-x86_64-{K80_DRIVER_VERSION}.run"
    else:
        run(f"curl -fSsl -O {DRIVER_URL}")
        binary = f"NVIDIA-Linux-x86_64-{DRIVER_VERSION}.run"

    attempt = 0
    no_drm = ""

    while attempt < 3:
        install_run = run(
            "sh {} -s {} {} --no-cc-version-check".format(binary, dkms, no_drm),
            check=False,
        )

        if install_run.returncode == 0:
            return

        if (
            "Failed to install the kernel module through DKMS"
            in install_run.stderr.decode()
        ):
            dkms = ""

        if "--no-drm" in install_run.stderr.decode():
            # Installer failed to install DRM KMS, so we try again with DRM disabled.
            no_drm = "--no-drm"

        attempt += 1