wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-06 03:22:33 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Update GPU detection logic in cuda_malloc.py

Browse Source 
Refactor GPU check logic to return True for NVIDIA GPUs.
This commit is contained in:
patientx 2025-12-22 02:18:29 +03:00 committed by GitHub
parent ce2cdd0161
commit 2edfcd6141
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
cuda_malloc.py
View File

@ -56,14 +56,11 @@ def cuda_malloc_supported():
except: except:
names = set() names = set()
for x in names: for x in names:
if "AMD" in x: if "NVIDIA" in x:
return False
elif "NVIDIA" in x:
for b in blacklist: for b in blacklist:
if b in x: if b in x:
return False return False
return False return True
#We don't need malloc at all with amd gpu's. So disabling all together.
version = "" version = ""