gpu info definition

This commit is contained in:
2025-09-10 05:10:51 +02:00
parent 838f1c737e
commit 68e87a5bee
2 changed files with 10 additions and 1 deletions

View File

@@ -1,3 +1,4 @@
from .model_manager import ModelManager
from gpu import get_gpu_info
__all__ = ["ModelManager"]
__all__ = ["ModelManager", "get_gpu_info"]

8
src/services/gpu.py Normal file
View File

@@ -0,0 +1,8 @@
from typing import Dict
def get_gpu_info() -> Dict:
# use the command "amdgpu_top -d -J"
# it will return the gpu information
# this function should just return all information of the one gpu it will find. we just expect it to be one gpu. no handling of multiple ones.
...