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__()[source]#

Print build and current running cluster configuration summary.

abstract is_built_with_cuda() bool[source]#

Check if the backend is built with CUDA.

abstract is_built_with_rocm() bool[source]#

Check if the backend is built with ROCm.

abstract get_compute_device() str[source]#

Get Compute device.

abstract get_ngpus() int[source]#

Get the number of GPUs.

get_backend_info() dict[source]#

Get backend information.