zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-24 10:03:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Limit Windows multi-GPU visibility (#15737)

Browse Source
This commit is contained in:
rattus
2026-08-20 09:31:47 +10:00
committed by GitHub
parent c67885b145
commit 5ab2f7a2d6
3 changed files with 48 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cuda_malloc.py
View File

@@ -28,19 +28,19 @@ def get_gpu_names():
device_info = DISPLAY_DEVICEA()
device_info.cb = ctypes.sizeof(device_info)
device_index = 0
gpu_names = set()
gpu_names = []
while user32.EnumDisplayDevicesA(None, device_index, ctypes.byref(device_info), 0):
device_index += 1
gpu_names.add(device_info.DeviceString.decode('utf-8'))
gpu_names.append(device_info.DeviceString.decode('utf-8'))
return gpu_names
return enum_display_devices()
else:
gpu_names = set()
gpu_names = []
out = subprocess.check_output(['nvidia-smi', '-L'])
for l in out.split(b'\n'):
if len(l) > 0:
gpu_names.add(l.decode('utf-8').split(' (UUID')[0])
gpu_names.append(l.decode('utf-8').split(' (UUID')[0])
return gpu_names
blacklist = {"GeForce GTX TITAN X", "GeForce GTX 980", "GeForce GTX 970", "GeForce GTX 960", "GeForce GTX 950", "GeForce 945M",