README.md 4.48 KB
Newer Older
Xiang Gao's avatar
Xiang Gao committed
1
# <img src=https://raw.githubusercontent.com/aiqm/torchani/master/logo1.png width=180/>  Accurate Neural Network Potential on PyTorch
Gao, Xiang's avatar
Gao, Xiang committed
2

3
4
Build:

Gao, Xiang's avatar
Gao, Xiang committed
5
6
7
8
9
[![flake8](https://zasdfgbnm.visualstudio.com/torchani/_apis/build/status/flake8)](https://zasdfgbnm.visualstudio.com/torchani/_build/latest?definitionId=3)
[![docs](https://zasdfgbnm.visualstudio.com/torchani/_apis/build/status/docs)](https://zasdfgbnm.visualstudio.com/torchani/_build/latest?definitionId=4)
[![runnable submodules](https://zasdfgbnm.visualstudio.com/torchani/_apis/build/status/runnable-submodules)](https://zasdfgbnm.visualstudio.com/torchani/_build/latest?definitionId=5)
[![unit tests](https://zasdfgbnm.visualstudio.com/torchani/_apis/build/status/unit-tests)](https://zasdfgbnm.visualstudio.com/torchani/_build/latest?definitionId=6)
[![tools](https://zasdfgbnm.visualstudio.com/torchani/_apis/build/status/tools)](https://zasdfgbnm.visualstudio.com/torchani/_build/latest?definitionId=7)
10
[![Python2 Inference](https://zasdfgbnm.visualstudio.com/torchani/_apis/build/status/python2?branchName=master)](https://zasdfgbnm.visualstudio.com/torchani/_build/latest?definitionId=11&branchName=master)
11
12
[![CodeFactor](https://www.codefactor.io/repository/github/aiqm/torchani/badge/master)](https://www.codefactor.io/repository/github/aiqm/torchani/overview/master)
[![codecov](https://codecov.io/gh/aiqm/torchani/branch/master/graph/badge.svg)](https://codecov.io/gh/aiqm/torchani)
Gao, Xiang's avatar
Gao, Xiang committed
13
[![Total alerts](https://img.shields.io/lgtm/alerts/g/aiqm/torchani.svg?logo=lgtm&logoWidth=18)](https://lgtm.com/projects/g/aiqm/torchani/alerts/)
Gao, Xiang's avatar
Gao, Xiang committed
14

Gao, Xiang's avatar
Gao, Xiang committed
15
Deploy (these builds only succeed on tagged commits):
16
17

[![Build Status](https://zasdfgbnm.visualstudio.com/torchani/_apis/build/status/Deploy%20docs?branchName=master)](https://zasdfgbnm.visualstudio.com/torchani/_build/latest?definitionId=9?branchName=master)
Gao, Xiang's avatar
Gao, Xiang committed
18
[![Build Status](https://zasdfgbnm.visualstudio.com/torchani/_apis/build/status/Deploy%20PYPI?branchName=master)](https://zasdfgbnm.visualstudio.com/torchani/_build/latest?definitionId=10?branchName=master)
19

Gao, Xiang's avatar
Gao, Xiang committed
20
TorchANI is a pytorch implementation of ANI. It is currently under alpha release, which means, the API is not stable yet. If you find a bug of TorchANI, or have some feature request, feel free to open an issue on GitHub, or send us a pull request.
Gao, Xiang's avatar
Gao, Xiang committed
21

Xiang Gao's avatar
Xiang Gao committed
22
23
<img src=https://raw.githubusercontent.com/aiqm/torchani/master/logo2.png width=500/>

Gao, Xiang's avatar
Gao, Xiang committed
24
25
# Install

26
TorchANI requires the latest preview version of PyTorch. You can install PyTorch by the following commands (assuming cuda10):
Gao, Xiang's avatar
Gao, Xiang committed
27

Richard Xue's avatar
Richard Xue committed
28
```bash
29
pip install numpy
30
pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html
Richard Xue's avatar
Richard Xue committed
31
```
32

33
34
If you updated TorchANI, you may also need to update PyTorch:

Richard Xue's avatar
Richard Xue committed
35
```bash
36
pip install --upgrade --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html
37
```
Gao, Xiang's avatar
Gao, Xiang committed
38

39
After installing the correct PyTorch, you can install TorchANI by:
Gao, Xiang's avatar
Gao, Xiang committed
40
41

```bash
42
pip install torchani
Gao, Xiang's avatar
Gao, Xiang committed
43
44
```

45
See also [PyTorch's official site](https://pytorch.org/get-started/locally/) for instructions of installing latest preview version of PyTorch.
Gao, Xiang's avatar
Gao, Xiang committed
46

47
48
Please install nightly PyTorch through `pip install` instead of `conda install`. If your PyTorch is installed through `conda install`, then `pip` would mistakenly recognize the package name as `torch` instead of `torch-nightly`, which would cause dependency issue when installing TorchANI.

49
50
51
52
53
54
To run the tests and examples, you must manually download a data package

```bash
./download.sh
```

Gao, Xiang's avatar
Gao, Xiang committed
55
56
57
58
59
60
61
62
# Paper

The original ANI-1 paper is:

* Smith JS, Isayev O, Roitberg AE. ANI-1: an extensible neural network potential with DFT accuracy at force field computational cost. Chemical science. 2017;8(4):3192-203.

We are planning a seperate paper for TorchANI, it will be available when we are ready for beta release of TorchANI.

63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
See also: [isayev/ASE_ANI](https://github.com/isayev/ASE_ANI)

# Develop

To install TorchANI from GitHub:

```bash
git clone https://github.com/aiqm/torchani.git
cd torchani
pip install -e .
```

After TorchANI has been installed, you can build the documents by running `sphinx-build docs build`. But make sure you
install dependencies:
```bash
pip install sphinx sphinx-gallery pillow matplotlib sphinx_rtd_theme
```

81
82
To manually run unit tests, do `python setup.py nosetests`

Gao, Xiang's avatar
Gao, Xiang committed
83
# Note to TorchANI developers
Gao, Xiang's avatar
Gao, Xiang committed
84
85
86
87
88

Never commit to the master branch directly. If you need to change something, create a new branch, submit a PR on GitHub.

You must pass all the tests on GitHub before your PR can be merged.

Gao, Xiang's avatar
Gao, Xiang committed
89
Code review is required before merging pull request.