Yuning You commited on
Commit
3218749
·
1 Parent(s): 9f7e00f
Files changed (1) hide show
  1. README.md +5 -0
README.md CHANGED
@@ -33,6 +33,11 @@ pip install scanpy e3nn
33
  pip install lightning==2.1.0
34
  pip install numpy==1.26.4
35
  ```
 
 
 
 
 
36
 
37
  ## Citation
38
  If you use this code for you research, please cite our paper.
 
33
  pip install lightning==2.1.0
34
  pip install numpy==1.26.4
35
  ```
36
+ where the compatibility between ```torch``` and ```torch-geometric``` is not always guaranteed (the first two lines), since these two guys are picky on the platform and the version of other packages. You may trial and error to find the right version of ```torch``` and ```torch-geometric``` that works for you, e.g., in some machine I once installed via:
37
+ ```
38
+ pip install torch==2.0.1 --index-url https://download.pytorch.org/whl/cu117
39
+ pip install torch-scatter torch-sparse torch-cluster torch-geometric -f https://data.pyg.org/whl/torch-2.0.1+cu117.html
40
+ ```
41
 
42
  ## Citation
43
  If you use this code for you research, please cite our paper.