perler commited on
Commit
ab0b54f
1 Parent(s): d4ba646

try cluster with url

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  pytorch-lightning>=2.1
2
  torch-geometric>=2
3
- pytorch-cluster>=1
4
  scikit-learn-intelex>=2024.1.0
5
  numpy>=1.26.4
6
  scikit-image>=0.22.0
 
1
  pytorch-lightning>=2.1
2
  torch-geometric>=2
3
+ torch-cluster -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
4
  scikit-learn-intelex>=2024.1.0
5
  numpy>=1.26.4
6
  scikit-image>=0.22.0