deepmd.tf.cluster.local

deepmd.tf.cluster.local#

Get local GPU resources.

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.

Module Contents#

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