update readme

theislab · Apr 30, 2024 · 574fd97 · 574fd97
1 parent d44166c
commit 574fd97
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -33,6 +33,10 @@ pip install geome
 1. Install the latest development version:
 
 ```bash
+mamba create -n geome  python=3.11
+mamba activate geome
+mamba install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia
+pip install torch-scatter torch-sparse torch-cluster -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
 pip install git+https://github.com/theislab/geome.git@main
 ```