def install_cudnn7()

in tools/ami-creator/scripts/win2019_cuda11_installer.py [0:0]

• 17 lines of code
• 5 McCabe index (conditional complexity)

def install_cudnn7():
    if os.path.exists("C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2\\bin\\cudnn64_7.dll"):
        logging.info("cuDNN7 already installed, skipping.")
        return False
    # cuDNN
    logging.info("Installing cuDNN7")
    with tempfile.TemporaryDirectory() as tmpdir:
        local_file = download(DEPS['cudnn7'])
        with zipfile.ZipFile(local_file, 'r') as zip:
            zip.extractall(tmpdir)
        for f in glob.glob(tmpdir+"\\cuda\\bin\\*"):
            copy(f, "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2\\bin")
        for f in glob.glob(tmpdir+"\\cuda\\include\\*.h"):
            copy(f, "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2\\include")
        for f in glob.glob(tmpdir+"\\cuda\\lib\\x64\\*"):
            copy(f, "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2\\lib\\x64")
    logging.info("cuDNN7 install complete")
    return True