gpu info definition
This commit is contained in:
@@ -1,3 +1,4 @@
|
|||||||
from .model_manager import ModelManager
|
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
8
src/services/gpu.py
Normal 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.
|
||||||
|
...
|
||||||
Reference in New Issue
Block a user