deepmd.tf.cluster.local

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()

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]

Get local resources: nodename, nodelist, and gpus.

Returns:

Tuple[str, List[str], Optional[List[int]]]

nodename, nodelist, and gpus