gpu info definition
This commit is contained in:
@@ -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
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