deepmd.tf.cluster

Module that reads node resources, auto detects if running local or on SLURM.

Submodules

• deepmd.tf.cluster.local

Functions

get_resource(→ tuple[str, list[str], list[int] | None])

Get local or slurm resources: nodename, nodelist, and gpus.

Package Contents

deepmd.tf.cluster.get_resource() → tuple[str, list[str], list[int] | None]

Get local or slurm resources: nodename, nodelist, and gpus.

Returns:

tuple[str, list[str], Optional[list[int]]]

nodename, nodelist, and gpus