manoskary commited on
Commit
718776a
·
verified ·
1 Parent(s): b57c33d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +19 -4
requirements.txt CHANGED
@@ -1,12 +1,23 @@
1
  # PyTorch (install first, required by other packages)
 
 
2
  torch>=2.0.0
3
  torchvision>=0.15.0
4
 
5
- # PyTorch extensions (require torch to be installed)
6
- pytorch-lightning>=2.0.0
7
- torch-geometric>=2.3.0
8
  torch-scatter>=2.1.0
9
  torch-sparse>=0.6.0
 
 
 
 
 
 
 
 
 
10
 
11
  # Data processing
12
  numpy>=1.21.0
@@ -34,5 +45,9 @@ torchmetrics>=0.11.0
34
  # Other utilities
35
  tqdm>=4.64.0
36
 
 
 
 
37
  # Core AnalysisGNN dependency (install last, after all dependencies)
38
- git+https://github.com/manoskary/analysisGNN.git
 
 
1
  # PyTorch (install first, required by other packages)
2
+ # Use CPU-only version for compatibility; change index URL if you need CUDA
3
+ --index-url https://download.pytorch.org/whl/cpu
4
  torch>=2.0.0
5
  torchvision>=0.15.0
6
 
7
+ # PyTorch Geometric extensions (installed without build isolation to use existing torch)
8
+ # Note: These may require compilation; use --no-build-isolation flag when installing
9
+ --no-build-isolation
10
  torch-scatter>=2.1.0
11
  torch-sparse>=0.6.0
12
+ torch-cluster>=1.6.0
13
+ torch-spline-conv>=1.2.0
14
+
15
+ # Reset to default PyPI index for remaining packages
16
+ --index-url https://pypi.org/simple
17
+
18
+ # PyTorch extensions
19
+ pytorch-lightning>=2.0.0
20
+ torch-geometric>=2.3.0
21
 
22
  # Data processing
23
  numpy>=1.21.0
 
45
  # Other utilities
46
  tqdm>=4.64.0
47
 
48
+ # Graph music representation library (required by AnalysisGNN)
49
+ git+https://github.com/manoskary/graphmuse.git
50
+
51
  # Core AnalysisGNN dependency (install last, after all dependencies)
52
+ # Note: The vendored copy in ./vendor/analysisGNN has been patched to support Python 3.12
53
+ -e ./vendor/analysisGNN