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

Auto-fix cpu-only torch installations on NVIDIA 5060

Browse Source
This commit is contained in:
cmdr2 2025-05-30 18:34:54 +05:30
parent 14e45bfd24
commit 1d27765150
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/check_modules.py
View File

@ -69,8 +69,9 @@ def update_modules():
else:
torch_version_str = version("torch")
torch_version = version_str_to_tuple(torch_version_str)
is_cpu_torch = "+" not in torch_version_str
print(f"Current torch version: {torch_version} ({torch_version_str})")
if torch_version < (2, 7):
if torch_version < (2, 7) or is_cpu_torch:
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):