def install_cudnn8()

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

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

def install_cudnn8():
    if os.path.exists("C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0\\bin\\cudnn64_8.dll"):
        logging.info("cuDNN7 already installed, skipping.")
        return False
    # cuDNN
    logging.info("Installing cuDNN8")
    with tempfile.TemporaryDirectory() as tmpdir:
        local_file = download(DEPS['cudnn8'])
        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\\v11.0\\bin")
        for f in glob.glob(tmpdir+"\\cuda\\include\\*.h"):
            copy(f, "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0\\include")
        for f in glob.glob(tmpdir+"\\cuda\\lib\\x64\\*"):
            copy(f, "C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0\\lib\\x64")
    logging.info("cuDNN8 install complete")
    return True