Skip to main content

A deep learning package for many-body potential energy representation and molecular dynamics

Project description

.. role:: raw-html-m2r(raw) :format: html

:raw-html-m2r:<span style="font-size:larger;">DeePMD-kit Manual</span>

.. image:: https://img.shields.io/github/release/deepmodeling/deepmd-kit.svg?maxAge=86400 :target: https://github.com/deepmodeling/deepmd-kit/releases :alt: GitHub release

.. image:: https://readthedocs.org/projects/deepmd/badge/?version=latest :target: https://deepmd.readthedocs.io/en/latest/?badge=latest :alt: Documentation Status

Table of contents

  • About DeePMD-kit <#about-deepmd-kit>_

    • Highlighted features <#highlighted-features>_
    • Code structure <#code-structure>_
    • License and credits <#license-and-credits>_
    • Deep Potential in a nutshell <#deep-potential-in-a-nutshell>_
  • Download and install <#download-and-install>_

  • Use DeePMD-kit <#use-deepmd-kit>_

  • Troubleshooting <#troubleshooting>_

About DeePMD-kit

DeePMD-kit is a package written in Python/C++, designed to minimize the effort required to build deep learning based model of interatomic potential energy and force field and to perform molecular dynamics (MD). This brings new hopes to addressing the accuracy-versus-efficiency dilemma in molecular simulations. Applications of DeePMD-kit span from finite molecules to extended systems and from metallic systems to chemically bonded systems.

For more information, check the documentation <https://deepmd.readthedocs.io/>_.

Highlighted features

  • interfaced with TensorFlow\ , one of the most popular deep learning frameworks, making the training process highly automatic and efficient.
  • interfaced with high-performance classical MD and quantum (path-integral) MD packages\ , i.e., LAMMPS and i-PI, respectively.
  • implements the Deep Potential series models\ , which have been successfully applied to finite and extended systems including organic molecules, metals, semiconductors, and insulators, etc.
  • implements MPI and GPU supports\ , makes it highly efficient for high performance parallel and distributed computing.
  • highly modularized\ , easy to adapt to different descriptors for deep learning based potential energy models.

Code structure

The code is organized as follows:

  • data/raw\ : tools manipulating the raw data files.

  • examples\ : example json parameter files.

  • source/3rdparty\ : third-party packages used by DeePMD-kit.

  • source/cmake\ : cmake scripts for building.

  • source/ipi\ : source code of i-PI client.

  • source/lib\ : source code of DeePMD-kit library.

  • source/lmp\ : source code of Lammps module.

  • source/op\ : tensorflow op implementation. working with library.

  • source/scripts\ : Python script for model freezing.

  • source/train\ : Python modules and scripts for training and testing.

License and credits

The project DeePMD-kit is licensed under GNU LGPLv3.0 <./LICENSE>_. If you use this code in any future publications, please cite this using Han Wang, Linfeng Zhang, Jiequn Han, and Weinan E. "DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamics." Computer Physics Communications 228 (2018): 178-184.

Deep Potential in a nutshell

The goal of Deep Potential is to employ deep learning techniques and realize an inter-atomic potential energy model that is general, accurate, computationally efficient and scalable. The key component is to respect the extensive and symmetry-invariant properties of a potential energy model by assigning a local reference frame and a local environment to each atom. Each environment contains a finite number of atoms, whose local coordinates are arranged in a symmetry preserving way. These local coordinates are then transformed, through a sub-network, to a so-called atomic energy. Summing up all the atomic energies gives the potential energy of the system.

The initial proof of concept is in the Deep Potential <http://www.global-sci.com/galley/CiCP-2017-0213.pdf>_ paper, which employed an approach that was devised to train the neural network model with the potential energy only. With typical ab initio molecular dynamics (AIMD) datasets this is insufficient to reproduce the trajectories. The Deep Potential Molecular Dynamics (\ DeePMD <https://journals.aps.org/prl/abstract/10.1103/PhysRevLett.120.143001>_\ ) model overcomes this limitation. In addition, the learning process in DeePMD improves significantly over the Deep Potential method thanks to the introduction of a flexible family of loss functions. The NN potential constructed in this way reproduces accurately the AIMD trajectories, both classical and quantum (path integral), in extended and finite systems, at a cost that scales linearly with system size and is always several orders of magnitude lower than that of equivalent AIMD simulations.

Although being highly efficient, the original Deep Potential model satisfies the extensive and symmetry-invariant properties of a potential energy model at the price of introducing discontinuities in the model. This has negligible influence on a trajectory from canonical sampling but might not be sufficient for calculations of dynamical and mechanical properties. These points motivated us to develop the Deep Potential-Smooth Edition (\ DeepPot-SE <https://arxiv.org/abs/1805.09003>_\ ) model, which replaces the non-smooth local frame with a smooth and adaptive embedding network. DeepPot-SE shows great ability in modeling many kinds of systems that are of interests in the fields of physics, chemistry, biology, and materials science.

In addition to building up potential energy models, DeePMD-kit can also be used to build up coarse-grained models. In these models, the quantity that we want to parameterize is the free energy, or the coarse-grained potential, of the coarse-grained particles. See the DeePCG paper <https://aip.scitation.org/doi/full/10.1063/1.5027645>_ for more details.

Download and install

Please follow our github <https://github.com/deepmodeling/deepmd-kit>_ webpage to download the latest released version <https://github.com/deepmodeling/deepmd-kit/tree/master>_ and development version <https://github.com/deepmodeling/deepmd-kit/tree/devel>_.

DeePMD-kit offers multiple installation methods. It is recommend using easily methods like offline packages <doc/install.md#offline-packages>\ , conda <doc/install.md#with-conda> and docker <doc/install.md#with-docker>_.

One may manually install DeePMD-kit by following the instuctions on installing the python interface <doc/install.md#install-the-python-interface>_ and installing the C++ interface <doc/install.md#install-the-c-interface>_. The C++ interface is necessary when using DeePMD-kit with LAMMPS and i-PI.

Use DeePMD-kit

The typical procedure of using DeePMD-kit includes 5 steps

#. Prepare data <doc/use-deepmd-kit.md#prepare-data>_ #. Train a model <doc/use-deepmd-kit.md#train-a-model>_ #. Freeze the model <doc/use-deepmd-kit.md#freeze-a-model>_ #. Test the model <doc/use-deepmd-kit.md#test-a-model>_ #. Inference the model in python <doc/use-deepmd-kit.md#model-inference>_ or using the model in other molecular simulation packages like LAMMPS <doc/use-deepmd-kit.md#run-md-with-lammps>\ , i-PI <doc/use-deepmd-kit.md#run-path-integral-md-with-i-pi> or ASE <doc/use-deepmd-kit.md#use-deep-potential-with-ase>_.

A quick-start on using DeePMD-kit can be found here <doc/use-deepmd-kit.md>_.

A full document <doc/train-input-auto.rst>_ on options in the training input script is available.

Troubleshooting

In consequence of various differences of computers or systems, problems may occur. Some common circumstances are listed as follows. If other unexpected problems occur, you're welcome to contact us for help.

Model compatability

When the version of DeePMD-kit used to training model is different from the that of DeePMD-kit running MDs, one has the problem of model compatability.

DeePMD-kit guarantees that the codes with the same major and minor revisions are compatible. That is to say v0.12.5 is compatible to v0.12.0, but is not compatible to v0.11.0 nor v1.0.0.

Installation: inadequate versions of gcc/g++

Sometimes you may use a gcc/g++ of version <4.9. If you have a gcc/g++ of version > 4.9, say, 7.2.0, you may choose to use it by doing

.. code-block:: bash

export CC=/path/to/gcc-7.2.0/bin/gcc export CXX=/path/to/gcc-7.2.0/bin/g++

If, for any reason, for example, you only have a gcc/g++ of version 4.8.5, you can still compile all the parts of TensorFlow and most of the parts of DeePMD-kit. i-Pi will be disabled automatically.

Installation: build files left in DeePMD-kit

When you try to build a second time when installing DeePMD-kit, files produced before may contribute to failure. Thus, you may clear them by

.. code-block:: bash

cd build rm -r *

and redo the cmake process.

MD: cannot run LAMMPS after installing a new version of DeePMD-kit

This typically happens when you install a new version of DeePMD-kit and copy directly the generated USER-DEEPMD to a LAMMPS source code folder and re-install LAMMPS.

To solve this problem, it suffices to first remove USER-DEEPMD from LAMMPS source code by

.. code-block:: bash

make no-user-deepmd

and then install the new USER-DEEPMD.

If this does not solve your problem, try to decompress the LAMMPS source tarball and install LAMMPS from scratch again, which typically should be very fast.

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

deepmd-kit-1.3.3.tar.gz (2.8 MB view details)

Uploaded Source

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

deepmd_kit-1.3.3-cp38-cp38-manylinux2010_x86_64.whl (476.1 kB view details)

Uploaded CPython 3.8manylinux: glibc 2.12+ x86-64

deepmd_kit-1.3.3-cp37-cp37m-manylinux2010_x86_64.whl (476.1 kB view details)

Uploaded CPython 3.7mmanylinux: glibc 2.12+ x86-64

deepmd_kit-1.3.3-cp36-cp36m-manylinux2010_x86_64.whl (476.1 kB view details)

Uploaded CPython 3.6mmanylinux: glibc 2.12+ x86-64

File details

Details for the file deepmd-kit-1.3.3.tar.gz.

File metadata

  • Download URL: deepmd-kit-1.3.3.tar.gz
  • Upload date:
  • Size: 2.8 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.1 importlib_metadata/3.7.3 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.9.2

File hashes

Hashes for deepmd-kit-1.3.3.tar.gz
Algorithm Hash digest
SHA256 623822bf7d5148c451bb6cdca86e2583c325482069323264ede9aa49a6199598
MD5 d026f3ef0f8a988f4d33eddf1a6aa380
BLAKE2b-256 77fe9a547548f33192078fd0268c12f3bc7c24670c4e8f799996b0ccf7dc1fa0

See more details on using hashes here.

File details

Details for the file deepmd_kit-1.3.3-cp38-cp38-manylinux2010_x86_64.whl.

File metadata

  • Download URL: deepmd_kit-1.3.3-cp38-cp38-manylinux2010_x86_64.whl
  • Upload date:
  • Size: 476.1 kB
  • Tags: CPython 3.8, manylinux: glibc 2.12+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.1 importlib_metadata/3.7.3 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.9.2

File hashes

Hashes for deepmd_kit-1.3.3-cp38-cp38-manylinux2010_x86_64.whl
Algorithm Hash digest
SHA256 95b638b42e6c9dd101bf8919f072347e6268053a3702ac75c1ea865918730cd4
MD5 aca3661b5c8bbfd88721924528fe8a8e
BLAKE2b-256 5ed477c30fdf031fda31bd20b9a70d26689beafa1c2eab1fccfb31f485c489aa

See more details on using hashes here.

File details

Details for the file deepmd_kit-1.3.3-cp37-cp37m-manylinux2010_x86_64.whl.

File metadata

  • Download URL: deepmd_kit-1.3.3-cp37-cp37m-manylinux2010_x86_64.whl
  • Upload date:
  • Size: 476.1 kB
  • Tags: CPython 3.7m, manylinux: glibc 2.12+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.1 importlib_metadata/3.7.3 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.9.2

File hashes

Hashes for deepmd_kit-1.3.3-cp37-cp37m-manylinux2010_x86_64.whl
Algorithm Hash digest
SHA256 3fb3baf9d8912bc73e1499cdf60fe708324c7a5f2da42759c83cbf7ec41513e9
MD5 7b15e58b3ccb3ef982230fed02726560
BLAKE2b-256 ac758a6f2319875c449e64c176929412d062e09c0e4b47f9cc13cfe9046d7da0

See more details on using hashes here.

File details

Details for the file deepmd_kit-1.3.3-cp36-cp36m-manylinux2010_x86_64.whl.

File metadata

  • Download URL: deepmd_kit-1.3.3-cp36-cp36m-manylinux2010_x86_64.whl
  • Upload date:
  • Size: 476.1 kB
  • Tags: CPython 3.6m, manylinux: glibc 2.12+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.1 importlib_metadata/3.7.3 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.9.2

File hashes

Hashes for deepmd_kit-1.3.3-cp36-cp36m-manylinux2010_x86_64.whl
Algorithm Hash digest
SHA256 b00b808c568d85a2b4f7ced86ab9d8bed05b349348543b37db18e37471aaa70e
MD5 f3a8792acc256a60991c10a9469f950a
BLAKE2b-256 a98c018b0442d3e155d263700a685ac38e14272a1937f11a51dbaf93657173a0

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page