rocmtop 0.1.0__tar.gz

rocmtop-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Ziyang Zhai (Liam)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

rocmtop-0.1.0/PKG-INFO

@@ -0,0 +1,51 @@
+Metadata-Version: 2.4
+Name: rocmtop
+Version: 0.1.0
+Summary: A lightweight ROCm GPU monitoring tool (like nvidia-smi top)
+License: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: rich
+Requires-Dist: readchar
+Dynamic: license-file
+
+# rocmtop
+
+**rocmtop** 是一个轻量级的 ROCm GPU 监控工具，可以在终端动态展示 AMD 显卡的状态（温度、功率、GPU 利用率和显存占用），类似于 `nvidia-smi top`。  
+
+用户可以直接通过终端使用：
+
+```bash
+pip install rocmtop
+
+rocmtop
+````
+
+## 功能特点
+
+* 实时刷新 GPU 状态（默认 0.5 秒刷新间隔）
+* 彩色显示 GPU 温度、功率、利用率和显存占用
+* 进度条直观显示 GPU 利用率
+* 按 `q` 或 `Q` 退出
+* 轻量、易用，无依赖复杂库，仅依赖 `rich` 和 `readchar`
+
+## 安装
+
+```bash
+pip install rocmtop
+```
+
+## 使用
+
+```bash
+rocmtop
+```
+
+> 注意：需要在支持 ROCm 的 AMD 显卡上使用，并安装 `rocm-smi` 工具。
+
+## 作者
+```
+Ziyang Zhai (Liam)
+
+```

rocmtop-0.1.0/README.md

@@ -0,0 +1,39 @@
+# rocmtop
+
+**rocmtop** 是一个轻量级的 ROCm GPU 监控工具，可以在终端动态展示 AMD 显卡的状态（温度、功率、GPU 利用率和显存占用），类似于 `nvidia-smi top`。  
+
+用户可以直接通过终端使用：
+
+```bash
+pip install rocmtop
+
+rocmtop
+````
+
+## 功能特点
+
+* 实时刷新 GPU 状态（默认 0.5 秒刷新间隔）
+* 彩色显示 GPU 温度、功率、利用率和显存占用
+* 进度条直观显示 GPU 利用率
+* 按 `q` 或 `Q` 退出
+* 轻量、易用，无依赖复杂库，仅依赖 `rich` 和 `readchar`
+
+## 安装
+
+```bash
+pip install rocmtop
+```
+
+## 使用
+
+```bash
+rocmtop
+```
+
+> 注意：需要在支持 ROCm 的 AMD 显卡上使用，并安装 `rocm-smi` 工具。
+
+## 作者
+```
+Ziyang Zhai (Liam)
+
+```

rocmtop-0.1.0/pyproject.toml

@@ -0,0 +1,19 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "rocmtop"
+version = "0.1.0"
+description = "A lightweight ROCm GPU monitoring tool (like nvidia-smi top)"
+readme = "README.md"
+requires-python = ">=3.8"
+license = { text = "MIT" }
+
+dependencies = [
+    "rich",
+    "readchar",
+]
+
+[project.scripts]
+rocmtop = "rocmtop.cli:main"

rocmtop-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

rocmtop-0.1.0/src/rocmtop/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

rocmtop-0.1.0/src/rocmtop/cli.py

@@ -0,0 +1,174 @@
+import subprocess
+import re
+import time
+from threading import Thread
+
+from rich.console import Console
+from rich.table import Table
+import readchar
+
+import argparse
+from . import __version__
+
+console = Console()
+exit_flag = False  # 按键退出标志
+
+REFRESH_INTERVAL = 0.5
+
+
+def parse_rocm_smi():
+    """解析 rocm-smi 输出为 GPU 信息列表"""
+    try:
+        output = subprocess.check_output(["rocm-smi"], text=True)
+    except Exception as e:
+        console.print(f"[red]Failed to run rocm-smi: {e}[/red]")
+        return []
+
+    gpus = []
+    lines = output.splitlines()
+
+    for line in lines:
+        line = line.strip()
+        if not line or line.startswith("=") or line.startswith("Device") or "Concise" in line:
+            continue
+
+        m_device = re.match(r'^(\d+)', line)
+        if not m_device:
+            continue
+        device = m_device.group(1)
+
+        percentages = re.findall(r'(\d+)%', line)
+        if len(percentages) >= 2:
+            vram = int(percentages[-2])
+            gpu_util = int(percentages[-1])
+        else:
+            vram = gpu_util = 0
+
+        temp_match = re.search(r'(\d+\.?\d*)°C', line)
+        temp = float(temp_match.group(1)) if temp_match else 0
+
+        power_match = re.search(r'(\d+\.?\d*)W', line)
+        power = float(power_match.group(1)) if power_match else 0
+
+        gpus.append({
+            "device": device,
+            "temp": temp,
+            "power": power,
+            "gpu_util": gpu_util,
+            "vram": vram
+        })
+
+    return gpus
+
+
+def color_gpu_util(val):
+    if val >= 90:
+        return "red"
+    elif val >= 50:
+        return "yellow"
+    elif val > 0:
+        return "green"
+    else:
+        return "white"
+
+
+def color_power(val):
+    if val > 600:
+        return "red"
+    elif val > 200:
+        return "yellow"
+    else:
+        return "green"
+
+
+def color_vram(val):
+    if val > 90:
+        return "red"
+    elif val >= 50:
+        return "yellow"
+    elif val > 0:
+        return "green"
+    else:
+        return "white"
+
+
+def color_temp(val):
+    if val > 90:
+        return "red"
+    elif val >= 60:
+        return "yellow"
+    elif val >= 30:
+        return "green"
+    else:
+        return "white"
+
+
+def display_gpus(gpus):
+    """美化显示 GPU 表格"""
+    table = Table(title="AMD GPU Monitor (rocmtop)", box=None, expand=True)
+    table.add_column("Device", style="bold cyan")
+    table.add_column("Temp (°C)", justify="right")
+    table.add_column("Power (W)", justify="right")
+    table.add_column("GPU Util", justify="right", width=30)
+    table.add_column("VRAM %", justify="right")
+
+    bar_len = 20
+
+    for gpu in gpus:
+        temp_str = f"[{color_temp(gpu['temp'])}]{gpu['temp']}[/]"
+        power_str = f"[{color_power(gpu['power'])}]{gpu['power']}[/]"
+
+        filled_len = int(gpu['gpu_util'] / 100 * bar_len)
+        bar = "█" * filled_len + " " * (bar_len - filled_len)
+        gpu_cell = (
+            f"[{color_gpu_util(gpu['gpu_util'])}]"
+            f"{bar}{str(gpu['gpu_util']).rjust(30 - len(bar))}%[/]"
+        )
+
+        vram_str = f"[{color_vram(gpu['vram'])}]{gpu['vram']}%[/]"
+
+        table.add_row(
+            gpu["device"],
+            temp_str,
+            power_str,
+            gpu_cell,
+            vram_str
+        )
+
+    console.clear()
+    console.print(table)
+    console.print("[bold magenta]Press 'q' or 'Q' to quit[/bold magenta]")
+
+
+def key_listener():
+    global exit_flag
+    while True:
+        c = readchar.readkey()
+        if c in ("q", "Q"):
+            exit_flag = True
+            break
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--version", action="store_true", help="Show version and exit")
+    args = parser.parse_args()
+
+    if args.version:
+        print(f"rocmtop version {__version__}")
+        return
+        
+    global exit_flag
+
+    Thread(target=key_listener, daemon=True).start()
+
+    while not exit_flag:
+        gpus = parse_rocm_smi()
+        if gpus:
+            display_gpus(gpus)
+        else:
+            console.print("[red]No GPU info found[/red]")
+        time.sleep(REFRESH_INTERVAL)
+
+    console.clear()
+    console.print("[bold green]Exiting rocmtop[/bold green]")

rocmtop-0.1.0/src/rocmtop.egg-info/PKG-INFO

@@ -0,0 +1,51 @@
+Metadata-Version: 2.4
+Name: rocmtop
+Version: 0.1.0
+Summary: A lightweight ROCm GPU monitoring tool (like nvidia-smi top)
+License: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: rich
+Requires-Dist: readchar
+Dynamic: license-file
+
+# rocmtop
+
+**rocmtop** 是一个轻量级的 ROCm GPU 监控工具，可以在终端动态展示 AMD 显卡的状态（温度、功率、GPU 利用率和显存占用），类似于 `nvidia-smi top`。  
+
+用户可以直接通过终端使用：
+
+```bash
+pip install rocmtop
+
+rocmtop
+````
+
+## 功能特点
+
+* 实时刷新 GPU 状态（默认 0.5 秒刷新间隔）
+* 彩色显示 GPU 温度、功率、利用率和显存占用
+* 进度条直观显示 GPU 利用率
+* 按 `q` 或 `Q` 退出
+* 轻量、易用，无依赖复杂库，仅依赖 `rich` 和 `readchar`
+
+## 安装
+
+```bash
+pip install rocmtop
+```
+
+## 使用
+
+```bash
+rocmtop
+```
+
+> 注意：需要在支持 ROCm 的 AMD 显卡上使用，并安装 `rocm-smi` 工具。
+
+## 作者
+```
+Ziyang Zhai (Liam)
+
+```

rocmtop-0.1.0/src/rocmtop.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+LICENSE
+README.md
+pyproject.toml
+src/rocmtop/__init__.py
+src/rocmtop/cli.py
+src/rocmtop.egg-info/PKG-INFO
+src/rocmtop.egg-info/SOURCES.txt
+src/rocmtop.egg-info/dependency_links.txt
+src/rocmtop.egg-info/entry_points.txt
+src/rocmtop.egg-info/requires.txt
+src/rocmtop.egg-info/top_level.txt

rocmtop-0.1.0/src/rocmtop.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

rocmtop-0.1.0/src/rocmtop.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+rocmtop = rocmtop.cli:main

rocmtop-0.1.0/src/rocmtop.egg-info/requires.txt

@@ -0,0 +1,2 @@
+rich
+readchar

rocmtop-0.1.0/src/rocmtop.egg-info/top_level.txt

@@ -0,0 +1 @@
+rocmtop