diff --git a/main.py b/main.py index aecb514..7d0b495 100644 --- a/main.py +++ b/main.py @@ -52,14 +52,14 @@ def get_server_gpu_status() -> list: { 'name': 'cml5' 'users': [ - 'snsd0805', 'snsd0805', 'timmy' + ['snsd0805', 'timmy'], ['timmy'] ], - 'gpu': 'V100' + 'gpus': ['V100', 'V100'] }, ... ] - means that 'cml5' server has 3 'V100', and the user 'snsd0805' is using 2 GPU and the user 'timmy' is usig 1 GPU now. + means that 'cml5' server has 2 'V100', and the user 'snsd0805' is using 1 GPU and the user 'timmy' is usig 2 GPU now. ''' servers = []