azalea/zshrc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[F] Fix multiple gpus

Browse Source
This commit is contained in:
Azalea Gui
2024-10-07 16:55:41 -04:00
parent 1c8a146b63
commit 6f6b88bd88
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/bin/tmux-tool.py
+2 -2
View File
@@ -67,14 +67,14 @@ def get_hardware_info(cpu=False, gpu=False):
if cpu:
# Fetch CPU info
cpu_info = subprocess.run([fastfetch, '--logo', 'none', '-s', 'CPU'], stdout=subprocess.PIPE)
cpu_str = cpu_info.stdout.decode().strip()
cpu_str = cpu_info.stdout.decode().splitlines()[0].strip()
# cpu_str = cpu_str.split(':', 1)[1].split('@', 1)[0].strip()
cpu_str = cpu_str.split(':', 1)[1].strip()
if gpu:
# Fetch GPU info
gpu_info = subprocess.run([fastfetch, '--logo', 'none', '-s', 'GPU'], stdout=subprocess.PIPE)
gpu_str = gpu_info.stdout.decode().strip()
gpu_str = gpu_info.stdout.decode().splitlines()[0].strip()
gpu_str = gpu_str.split(':', 1)[1].strip()
gpu_str = re.sub(r'lite hash rate', 'LHR', gpu_str, flags=re.IGNORECASE)