deepmd.tf.cluster package

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

deepmd.tf.cluster.get_resource() → Tuple[str, List[str], Optional[List[int]]]

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

Returns

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

nodename, nodelist, and gpus

Submodules

deepmd.tf.cluster.local module

Get local GPU resources.

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], Optional[List[int]]]

Get local resources: nodename, nodelist, and gpus.

Returns

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

nodename, nodelist, and gpus