Skip to content

Commit 9c26435

Browse files
committed
doc: rephrase torch install section
1 parent 28a272d commit 9c26435

1 file changed

Lines changed: 7 additions & 11 deletions

File tree

docs/source/install/installing_mala.rst

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,16 @@ Prerequisites
77
MALA supports any Python version starting from ``3.10.4``. No upper limit on
88
Python versions are enforced. The most recent *tested* version is ``3.10.12``.
99

10-
MALA requires ``torch`` (https://pytorch.org). Most install sites will want to
11-
use an already installed ``torch`` or install a specific version before
12-
installing the MALA package itself, for instance
13-
14-
.. code-block:: sh
15-
16-
pip install --index-url https://download.pytorch.org/whl/cu<cuda_version> torch==<torch_version>
17-
18-
If no ``torch`` is found, we install the latest GPU-enabled version via ``pip
19-
install torch`` (see ``requirements.txt``).
20-
2110
Installing the Python library
2211
*****************************
2312

13+
MALA requires ``torch`` (https://pytorch.org). We install the latest
14+
GPU-enabled version (see ``requirements.txt``),
15+
unless you have ``torch`` already installed (for example a version that supports
16+
AMD's ROCm a specific CUDA version).
17+
18+
To install the MALA package
19+
2420
* Download the MALA repository, e.g., with ``git clone git@github.com:mala-project/mala.git``
2521
* Change into the directory you cloned the repository to
2622
* Install MALA via ``pip install -e .[options]``

0 commit comments

Comments
 (0)