start.rst 1.1 KB
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
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
    pip install numpy torchvision_nightly
    pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html

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

.. code-block:: bash

    pip install --upgrade torch_nightly -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html


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.

.. warning::

    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.

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