deepmd.tf.lmp

Register entry points for lammps-wheel.

Module Contents

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.

Attributes

find_libpython
lib_env
preload_env
tf_dir
op_dir
cuda_library_paths
libpython

deepmd.tf.lmp.find_libpython

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

Get the environment variable from given paths.

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

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'

deepmd.tf.lmp.preload_env = 'LD_PRELOAD'

deepmd.tf.lmp.tf_dir

deepmd.tf.lmp.op_dir

deepmd.tf.lmp.cuda_library_paths = []

deepmd.tf.lmp.libpython

deepmd.tf.lmp.get_op_dir() → str

Get the directory of the deepmd-kit OP library.