extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-17 15:27:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

Tell the user to reinstall ED if they're trying to use a 50xx GPU with Python 3.8-based Easy Diffusion

Browse Source
This commit is contained in:
cmdr2 2025-04-27 21:28:56 +05:30
parent adfe24fdd0
commit 40801de615
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/check_modules.py
View File

@ -74,6 +74,15 @@ def update_modules():
gpu_infos = get_gpus() gpu_infos = get_gpus()
device_names = set(gpu.device_name for gpu in gpu_infos) device_names = set(gpu.device_name for gpu in gpu_infos)
if any(BLACKWELL_DEVICES.search(device_name) for device_name in device_names): if any(BLACKWELL_DEVICES.search(device_name) for device_name in device_names):
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") print("Upgrading torch to support NVIDIA 50xx series of graphics cards")
torchruntime.install(["--force", "--upgrade", "torch", "torchvision"]) torchruntime.install(["--force", "--upgrade", "torch", "torchvision"])