start.rst 797 Bytes
Newer Older
Gao, Xiang's avatar
Gao, Xiang committed
1
2
3
Installation
============

4
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
5

6
.. code-block:: bash
7

8
9
    pip install numpy
    pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html
10
11
12
13
14

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

.. code-block:: bash

15
    pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html
16
17
18
19
20
21
22
23
24
25
26


After installing the correct PyTorch, you can install TorchANI by:

.. code-block:: bash

    pip install torchani

See also `PyTorch's official site`_ for instructions of installing latest preview version of PyTorch.

.. _PyTorch's official site:
27
    https://pytorch.org/get-started/locally/