deepmd.tf.cluster

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

Submodules

• deepmd.tf.cluster.local

Package Contents

Functions

get_resource(→ Tuple[str, List[str], Optional[List[int]]])

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

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