deepmd.tf.lmp

Register entry points for lammps-wheel.

Attributes

find_libpython

lib_env

preload_env

tf_dir

op_dir

cuda_library_paths

libpython

Functions

get_env(→ str)

Get the environment variable from given paths.

get_library_path(→ List[str])

Get library path from a module.

get_op_dir(→ str)

Get the directory of the deepmd-kit OP library.

Module Contents

deepmd.tf.lmp.find_libpython = None[source]
deepmd.tf.lmp.get_env(paths: List[str | None]) → str[source]

Get the environment variable from given paths.

deepmd.tf.lmp.get_library_path(module: str, filename: str) → List[str][source]

Get library path from a module.

Parameters:
modulestr

The module name.

filenamestr

The library filename pattern.

Returns:
list[str]

The library path.

deepmd.tf.lmp.lib_env = 'LD_LIBRARY_PATH'[source]
deepmd.tf.lmp.preload_env = 'LD_PRELOAD'[source]
deepmd.tf.lmp.tf_dir[source]
deepmd.tf.lmp.op_dir[source]
deepmd.tf.lmp.cuda_library_paths = [][source]
deepmd.tf.lmp.libpython[source]
deepmd.tf.lmp.get_op_dir() → str[source]

Get the directory of the deepmd-kit OP library.