From f9c664dc3ce3930fd4f978dc20058b62bdbcee29 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Sun, 27 Apr 2025 21:28:56 +0530 Subject: [PATCH] Tell the user to reinstall ED if they're trying to use a 50xx GPU with Python 3.8-based Easy Diffusion --- scripts/check_modules.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/scripts/check_modules.py b/scripts/check_modules.py index afddd729..5d25bf7e 100644 --- a/scripts/check_modules.py +++ b/scripts/check_modules.py @@ -74,8 +74,17 @@ def update_modules(): gpu_infos = get_gpus() device_names = set(gpu.device_name for gpu in gpu_infos) if any(BLACKWELL_DEVICES.search(device_name) for device_name in device_names): - print("Upgrading torch to support NVIDIA 50xx series of graphics cards") - torchruntime.install(["--force", "--upgrade", "torch", "torchvision"]) + if sys.version_info < (3, 9): + print( + "\n###################################\n" + "NVIDIA 50xx series of graphics cards detected!\n\n" + "To use this graphics card, please install the latest version of Easy Diffusion from: https://github.com/easydiffusion/easydiffusion#installation" + "\n###################################\n" + ) + sys.exit() + else: + print("Upgrading torch to support NVIDIA 50xx series of graphics cards") + torchruntime.install(["--force", "--upgrade", "torch", "torchvision"]) for module_name, allowed_versions in modules_to_check.items(): if os.path.exists(f"src/{module_name}"):