vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2025-09-10 17:14:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

get gpu names with vulkaninfo

Browse source
This commit is contained in:
Concedo 2024-01-26 12:58:30 +08:00
parent c9ecd0f7c6
commit 87d852b85c
1 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

16
koboldcpp.py
View file

@ -1146,6 +1146,7 @@ def show_new_gui():
CUDevices = ["1","2","3","4","All"] CUDevices = ["1","2","3","4","All"]
CLDevicesNames = ["","","",""] CLDevicesNames = ["","","",""]
CUDevicesNames = ["","","","",""] CUDevicesNames = ["","","","",""]
VKDevicesNames = ["","","",""]
MaxMemory = [0] MaxMemory = [0]
tabcontent = {} tabcontent = {}
@ -1360,6 +1361,17 @@ def show_new_gui():
except Exception as e: except Exception as e:
pass pass
try: # Get Vulkan names
output = run(['vulkaninfo','--summary'], capture_output=True, text=True, check=True, encoding='utf-8').stdout
devicelist = [line.split("=")[1].strip() for line in output.splitlines() if "deviceName" in line]
idx = 0
for dname in devicelist:
if idx<len(VKDevicesNames):
VKDevicesNames[idx] = dname
idx += 1
except Exception as e:
pass
for idx in range(0,4): for idx in range(0,4):
if(len(FetchedCUdevices)>idx): if(len(FetchedCUdevices)>idx):
CUDevicesNames[idx] = FetchedCUdevices[idx] CUDevicesNames[idx] = FetchedCUdevices[idx]
@ -1461,8 +1473,8 @@ def show_new_gui():
s = int(gpu_choice_var.get())-1 s = int(gpu_choice_var.get())-1
v = runopts_var.get() v = runopts_var.get()
if v == "Use Vulkan": if v == "Use Vulkan":
quick_gpuname_label.configure(text="") quick_gpuname_label.configure(text=VKDevicesNames[s])
gpuname_label.configure(text="") gpuname_label.configure(text=VKDevicesNames[s])
elif v == "Use CLBlast" or v == "CLBlast NoAVX2 (Old CPU)": elif v == "Use CLBlast" or v == "CLBlast NoAVX2 (Old CPU)":
quick_gpuname_label.configure(text=CLDevicesNames[s]) quick_gpuname_label.configure(text=CLDevicesNames[s])
gpuname_label.configure(text=CLDevicesNames[s]) gpuname_label.configure(text=CLDevicesNames[s])