Get local GPU resources.Module Contents
Functions
get_gpus ()
| Get available IDs of GPU cards at local. |
get_resource (→ Tuple[str, List[str], Optional[List[int]]])
| Get local resources: nodename, nodelist, and gpus. |
- deepmd.tf.cluster.local.get_gpus()[source]
Get available IDs of GPU cards at local. These IDs are valid when used as the TensorFlow device ID.
- Returns:
Optional
[List
[int
]]List of available GPU IDs. Otherwise, None.
- deepmd.tf.cluster.local.get_resource() → Tuple[str, List[str], List[int] | None][source]
Get local resources: nodename, nodelist, and gpus.
- Returns:
Tuple
[str
, List
[str
], Optional
[List
[int
]]]nodename, nodelist, and gpus