  import json
  import locale
  import os
  import platform
  import re
  import sys
  from subprocess import PIPE, Popen
  
  import psutil
  
 
</s> remove class _SysInfo():
 </s> add class _SysInfo():  # pylint:disable=too-few-public-methods </s> remove             self.output.warning("Couldn't install {} with pip. "
                                "Please install this package manually".format(package))
 </s> add             self.output.warning(f"Couldn't install {package} with pip. "
                                "Please install this package manually") </s> remove         self.env.cuda_version = cuda_keys[0].lower().replace("cuda_path_v", "").replace("_", ".")
        self.env.cuda_path = os.environ[cuda_keys[0]]
        self.output.info("CUDA version: " + self.env.cuda_version)

    def cudnn_check(self):
        """ Check Linux or Windows cuDNN Version from cudnn.h """
        if self.env.os_version[0] == "Linux":
            cudnn_checkfiles = self.cudnn_checkfiles_linux()
        elif self.env.os_version[0] == "Windows":
            if not self.env.cuda_path and not self.cuda_keys_windows:
                self.output.error(
                    "CUDA not found. See "
                    "https://github.com/deepfakes/faceswap/blob/master/INSTALL.md#cuda "
                    "for instructions")
                return
            if not self.env.cuda_path:
                self.env.cuda_path = os.environ[self.cuda_keys_windows[0]]
            cudnn_checkfiles = self.cudnn_checkfiles_windows()

        cudnn_checkfile = None
        for checkfile in cudnn_checkfiles:
            if os.path.isfile(checkfile):
                cudnn_checkfile = checkfile
                break
 </s> add     def _cudnn_check(self):
        """ Check Linux or Windows cuDNN Version from cudnn.h and add to :attr:`cudnn_version`. """
        cudnn_checkfiles = getattr(self, f"_get_checkfiles_{self._os}")()
        cudnn_checkfile = next((hdr for hdr in cudnn_checkfiles if os.path.isfile(hdr)), None)
  import json
  import locale
  import os
  import platform

  import sys
  from subprocess import PIPE, Popen
  
  import psutil
  
keep keep keep keep replace keep keep keep keep keep
 sysinfo/setup - Standardize cuda/cudnn checks 
lib/sysinfo.py