deepmd.utils.summary

Module Contents

Classes

SummaryPrinter

Base summary printer.

Attributes

log

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.