Skip to content

Commit b3e3b2c

Browse files
authored
Update README.md
1 parent f381139 commit b3e3b2c

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,9 @@ Ilyes Batatia, Gregor Simm, and David Kovacs.
77

88
Requirements:
99
* Python >= 3.7
10-
* [PyTorch](https://pytorch.org/) >= 1.8
10+
* [PyTorch](https://pytorch.org/) >= 1.12
11+
12+
(for openMM, use Python = 3.9)
1113

1214
### conda installation
1315

@@ -36,8 +38,8 @@ To install via `pip`, follow the steps below:
3638
python -m venv mace-venv
3739
source mace-venv/bin/activate
3840

39-
# Install PyTorch (for example, for CUDA 10.2 [cu102])
40-
pip install torch==1.8.2 --extra-index-url "https://download.pytorch.org/whl/lts/1.8/cu102"
41+
# Install PyTorch (for example, for CUDA 11.6 [cu116])
42+
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
4143

4244
# Clone and install MACE (and all required packages)
4345
git clone git@github.com:ACEsuit/mace.git

0 commit comments

Comments
 (0)