deepmd.utils.summary

Contents

deepmd.utils.summary#

Attributes#

Classes#

SummaryPrinter

Base summary printer.

Module Contents#

deepmd.utils.summary.log[source]#

class deepmd.utils.summary.SummaryPrinter[source]#

Bases: abc.ABC

Base summary printer.

Backends should inherit from this class and implement the abstract methods.

WELCOME = (' _____ _____ __ __ _____ _ _ _ ', '| __ \\ | ...[source]#

CITATION = ('Please read and cite:', 'Wang, Zhang, Han and E, Comput.Phys.Comm. 228, 178-184 (2018)', 'Zeng...[source]#

BUILD: ClassVar[source]#

__call__() → None[source]#: Print build and current running cluster configuration summary.

abstractmethod is_built_with_cuda() → bool[source]#: Check if the backend is built with CUDA.

abstractmethod is_built_with_rocm() → bool[source]#: Check if the backend is built with ROCm.

abstractmethod get_compute_device() → str[source]#: Get Compute device.

abstractmethod get_ngpus() → int[source]#: Get the number of GPUs.

abstractmethod get_device_name() → str | None[source]#: Get the device name (e.g., NVIDIA A800-SXM4-80GB) if available.

get_backend_info() → dict[source]#: Get backend information.