def detect_gpu_device()

in linux/cuda_installer/os_installers/__init__.py [0:0]

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

    def detect_gpu_device(cls) -> Optional[str]:
        """
        Check if there is an NVIDIA GPU device attached and return its device code.
        """
        lspci = cls.run("lspci -n", silent=True)
        output = lspci.stdout
        dev_re = re.compile(r"10de:[\w\d]{4}")
        for line in output.splitlines():
            dev_code = dev_re.findall(line)
            if dev_code:
                return dev_code[0]
        else:
            return None